zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-21 20:46:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

adding helium into training

Browse Source
This commit is contained in:
Goekdeniz-Guelmez 2025-01-18 20:52:00 +01:00
parent 0e9fd16b26
commit 84055470ab
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
llms/mlx_lm/tuner/utils.py
View File

@ -94,6 +94,7 @@ def linear_to_lora_layers(
"phimoe", "phimoe",
"gemma", "gemma",
"gemma2", "gemma2",
"helium",
"starcoder2", "starcoder2",
"cohere", "cohere",
"cohere2", "cohere2",