mlx-examples/llms/tests
Anchen 949f63f309
chore(mlx-lm): fix print_trainable_parameters for quant models (#581)
* chore(mlx-lm): fix print_trainable_parameters for quant models

* chore: clean up

* refactor: use layer type to check quant bits

* chore: address comment
2024-03-20 08:41:03 -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 Add support for Cohere's Command-R (#565) 2024-03-13 07:03:36 -07:00
test_utils.py LoRA on all linear transformer block layers (#546) 2024-03-12 07:37:40 -07:00