mlx-examples

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 21:01:32 +08:00

Files

Shunta Saito c37e26a1a3 Add plamo-2-1b model (#1283 )

* Add pfnet/plamo-2-1b

* Fix cache.py to support non-top level layers

* Use mlx's BaseModelArgs

* Fix model

* Use sanitize()

* Remove unnecessary changes

* Add plamo2.py

* Apply formatter

* Fix some part

* Allow a cache obj defined externally

* Fix channel first weights to channel last for right use of MLX's conv1d

* Remove unused code part

* Give all inputs when it's the first time call of model

* Fix import

* Include .jsonl files to download from Huggingface hub

* Fix reference to layers

* Remove unnecessary code and add a test for plamo2

* Do not pass mask to prepare_inputs_for_generation

* Fix to use repeat instead of tile

* Add state property to PlamoCache

* Add __iter__ and __next__ methods to PlamoCache

* cleanup

* cleanup

* fix

---------

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

2025-02-24 19:24:43 -08: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 )