mlx-examples/llms/mlx_lm/tuner
Chime Ogbuji cb87f6f22c Add response template (or token) argument
For use in calculating mask for everything up to the after the response prompt (i.e., the continuation/completion)
2025-02-09 07:43:01 -08:00
..
__init__.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
datasets.py Add response template (or token) argument 2025-02-09 07:43:01 -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 Synch use of special tokens with iterate_batches 2025-02-09 07:41:24 -08:00
utils.py Add IBM granite model (#1265) 2025-02-08 15:46:15 -08:00