Merge branch 'main' into adding-support-for-mamba2

2025-12-16 02:08:55 +08:00 · 2025-01-26 16:58:06 +01:00
parent 2462a34194 77faa14ba4
commit de856c7223
3 changed files with 185 additions and 1 deletions
--- a/llms/mlx_lm/tuner/utils.py
+++ b/llms/mlx_lm/tuner/utils.py
@@ -94,6 +94,7 @@ def linear_to_lora_layers(
        "phimoe",
        "gemma",
        "gemma2",
+        "helium",
        "starcoder2",
        "cohere",
        "cohere2",