mlx-examples/llms/mlx_lm/tuner
Awni Hannun 657b4cc0aa
[MLX LM] Sampler refactor + a few improvements (#1094)
* starting

* refactor sampler/processor and a few improvements

* fix stream

* fix stream generate

* fix eos handling in stream generate
2024-11-07 16:15:24 -08: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 [MLX LM] Sampler refactor + a few improvements (#1094) 2024-11-07 16:15:24 -08:00
utils.py Adding full finetuning (#903) 2024-09-29 17:12:47 -07:00