3 people 87ddbfea30 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 <[email protected]>

2024-06-02 16:33:20 -07:00

..

test_datsets.py

Support for OpenAI’s fine-tuning dataset format (#548)

2024-03-19 16:45:46 -07:00

test_gguf.py

fix(mlx-lm): type hints in gguf.py (#621)

2024-03-26 07:56:01 -07:00

test_lora.py

LoRA: Extract small function (#614)

2024-06-02 06:38:42 -07:00

test_models.py

GPT2 Support (#798)

2024-06-02 16:33:20 -07:00

test_sample_utils.py

fix(mlx-lm): type hints in gguf.py (#621)

2024-03-26 07:56:01 -07:00

test_server.py

fix(mlx-lm): broken server.py (#690)

2024-04-18 14:26:18 -07:00

test_tuner_utils.py

LoRA: Extract small function (#614)

2024-06-02 06:38:42 -07:00

test_utils.py

LoRA on all linear transformer block layers (#546)

2024-03-12 07:37:40 -07:00