mlx-examples/llms/mlx_lm/tuner
Awni Hannun 2146bcd7ee
Quantize embedding / Update quantize API (#680)
* more async eval

* quantize embedding / update quantize api

* more updates for quantize

* update for quantize embeddings

* update sd quant API

* update sdxl quants

* error for datasets < batch_size

* async

* fix config loading

* fix quant

* fix tests

* fix req

* remove lm head if tie weights is true

* fix test
2024-04-18 18:16:10 -07:00
..
__init__.py feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
datasets.py Support for OpenAI’s fine-tuning dataset format (#548) 2024-03-19 16:45:46 -07:00
lora.py cast around lora adapters (#613) 2024-03-24 19:34:51 -07:00
trainer.py Quantize embedding / Update quantize API (#680) 2024-04-18 18:16:10 -07:00
utils.py Save lora config (#636) 2024-04-02 13:52:53 -07:00