mlx-examples/llms/mlx_lm/tuner
Zai Thottakath 418d9a5511
Feature: QDoRA (#891)
* feat: QDoRA with tests and a small bug fix for recalculation of self.m

* some simplifications and fixes

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-09-30 08:01:11 -07:00
..
__init__.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
datasets.py LoRA: Support HuggingFace dataset via data parameter (#996) 2024-09-30 07:36:21 -07: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 Adding full finetuning (#903) 2024-09-29 17:12:47 -07:00
utils.py Adding full finetuning (#903) 2024-09-29 17:12:47 -07:00