mlx-examples/llms/mlx_lm/models/gpt_bigcode.py at openlm

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00

Files

Prince Canuma b044ce2acf Add support for ibm granite (#758 )

* add support for granite 3-8B config

* add gpt_bigcode

* add positional embedding condition.

* add support for granite 3-8B config

* add gpt_bigcode

* add positional embedding condition.

* remove unused function

* rebase fix

* move position emebedding to mask creation

* add to tuner and format

* add support for granite 3-8B config

* add gpt_bigcode

* add positional embedding condition.

* add support for granite 3-8B config

* add gpt_bigcode

* add positional embedding condition.

* rebase fix

* move position emebedding to mask creation

* add to tuner and format

* refactor mask

* remove dropout layers

2024-05-21 20:16:31 -07:00

5.6 KiB

Raw Permalink Blame History

View Raw

5.6 KiB Raw Permalink Blame History

5.6 KiB

Raw Permalink Blame History