mlx-examples/llms/mlx_lm/tuner
Chime Ogbuji 0228c46434
Custom local dataset features (#1085)
* Generalize prompt_feature and completion_feature for use in local datasets to facilitate compatibility with many other training dataset formats.

* Persist configured prompt/completion key

* rebase + nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2025-01-13 10:01:18 -08:00
..
__init__.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
datasets.py Custom local dataset features (#1085) 2025-01-13 10:01:18 -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 fix encoding with special tokens + chat template (#1189) 2025-01-03 10:50:59 -08:00
utils.py Add support for cohere2 (#1157) 2024-12-16 08:01:03 -08:00