mlx-examples/llms/mlx_lm/tuner
Gökdeniz Gülmez 77faa14ba4
adding support for kyutai's helium (#1208)
* initial commit

* adding helium into training

* Update ACKNOWLEDGMENTS.md

* nits

* nits

* fixes / nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2025-01-26 07:19:07 -08:00
..
__init__.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
datasets.py Fix dataset variable name, in datasets.py (#1212) 2025-01-21 14:12:43 -08:00
dora.py Feature: QDoRA (#891) 2024-09-30 08:01:11 -07:00
lora.py Allow the entire model to be targed for LoRA and DoRA fine tuning: LoRA and DoRA embeddings with small DoRALinear bug fix (#914) 2024-08-16 07:38:36 -07:00
trainer.py reduction moved to CPU in case of distributed training (#1200) 2025-01-14 17:20:42 -08:00
utils.py adding support for kyutai's helium (#1208) 2025-01-26 07:19:07 -08:00