diff --git a/llms/mlx_lm/tuner/utils.py b/llms/mlx_lm/tuner/utils.py index 3986952a..9df9ba0d 100644 --- 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",