add to tuner

2025-08-29 01:46:09 +08:00 · 2024-12-14 16:39:55 +01:00 · 2024-12-14 16:39:55 +01:00 · 0337646b4e
commit 0337646b4e
parent 2f443cc6d7
1 changed files with 1 additions and 0 deletions
--- a/llms/mlx_lm/tuner/utils.py
+++ b/llms/mlx_lm/tuner/utils.py
@ -96,6 +96,7 @@ def linear_to_lora_layers(
        "gemma2",
        "starcoder2",
        "cohere",
+        "cohere2",
        "minicpm",
        "deepseek",
        "olmo2",