mlx-examples

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-24 15:58:11 +08:00

Files

Prince Canuma 2fce02acd8 Add support for Gemma3 (#1336 )

* add support for gemma3

* fix model loading

* revert rmsnorm

* revert is sliding pattern

* revert

* add tests

* formatting

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* fix sliding window mask

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>

2025-03-13 08:14:25 -07:00

test_datsets.py

Completion only fine-tuning of instruction models with collections of HF datasets (#1103 )

2025-02-09 20:12:34 -08:00

test_finetune.py

reduction moved to CPU in case of distributed training (#1200 )

2025-01-14 17:20:42 -08:00

test_generate.py

Add "from_draft" to GenerationResponse (#1272 )

2025-02-11 15:41:02 -08:00

test_gguf.py

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