Awni Hannun b8a348c1b8 Switch to fast RMS/LN Norm (#603) ...

* use nn.RMSNorm, use sdpa, cleanup

* bump mlx versions

* minor update

* use fast layer norm

* version bump

* update requirement for whisper

* update requirement for gguf

2024-03-23 07:13:51 -07:00

..

test_datsets.py

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

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

test_lora.py

chore(mlx-lm): fix print_trainable_parameters for quant models (#581)

2024-03-20 08:41:03 -07:00

test_models.py

Switch to fast RMS/LN Norm (#603)

2024-03-23 07:13:51 -07:00

test_sample_utils.py

chore(mlx-lm): fix the top_p implementation. (#602)

2024-03-21 12:18:23 -07:00

test_utils.py

LoRA on all linear transformer block layers (#546)

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