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)	2024-03-26 07:56:01 -07:00
test_models.py	Add plamo-2-1b model (#1283)	2025-02-24 19:24:43 -08:00
test_prompt_cache.py	Allow prompt callback to generate_step (#1133)	2024-12-03 16:17:14 -08:00
test_sample_utils.py	batched min p and fix spec gen sampling (#1222)	2025-01-27 15:40:31 -08:00
test_server.py	chore(mlx-lm): support text type content in messages (#1225)	2025-01-27 17:13:50 -08:00
test_tokenizers.py	Fix decoding manually added tokens (#1164)	2024-12-17 09:54:29 -08:00
test_tuner_utils.py	LoRA: Extract small function (#614)	2024-06-02 06:38:42 -07:00
test_utils_load_model.py	deepseek v3 model with pipeline parallelism (#1191)	2025-01-09 15:55:53 -08:00
test_utils.py	Fix whipser conversion for safetensors models (#935)	2024-08-14 10:22:04 -07:00