adding helium into training

2025-08-21 20:46:50 +08:00 · 2025-01-18 20:52:00 +01:00 · 2025-01-18 20:52:00 +01:00 · 84055470ab
commit 84055470ab
parent 0e9fd16b26
1 changed files with 1 additions and 0 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",