Prince Canuma c012eb173f Add support for OpenELM (#719) ...

* add openELM

* update splitting logic

* update qkv logic and, transformer and MLP block

* code formatting and fix args

* fix array slicing and remove unused var :)

* add to tuner

* use mx.split for slicing qkv

* merge with phi3

* remove rope scaling logic

* code formatting

2024-04-25 16:49:28 -07:00

..

__init__.py

feat: move lora into mlx-lm (#337)

2024-01-23 08:44:37 -08:00

datasets.py

Support for OpenAI’s fine-tuning dataset format (#548)

2024-03-19 16:45:46 -07:00

lora.py

cast around lora adapters (#613)

2024-03-24 19:34:51 -07:00

trainer.py

Quantize embedding / Update quantize API (#680)

2024-04-18 18:16:10 -07:00

utils.py

Add support for OpenELM (#719)

2024-04-25 16:49:28 -07:00