mlx-examples/llms/mlx_lm/tuner
Derek Lewis 89b0b75250
GPT2 Support (#798)
* GPT-2 model support

* Add test for gpt2 model

* Fix weight sanitizing for quantization

* use approx gelu

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-06-02 16:33:20 -07:00
..
__init__.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
datasets.py Support for OpenAI’s fine-tuning dataset format (#548) 2024-03-19 16:45:46 -07:00
dora.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
lora.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
trainer.py LoRA: Extract small function (#614) 2024-06-02 06:38:42 -07:00
utils.py GPT2 Support (#798) 2024-06-02 16:33:20 -07:00