zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 16:58:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add to tuner

Browse Source
This commit is contained in:
Prince Canuma 2024-12-14 16:39:55 +01:00
parent 2f443cc6d7
commit 0337646b4e
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

@ -96,6 +96,7 @@ def linear_to_lora_layers(
"gemma2", "gemma2",
"starcoder2", "starcoder2",
"cohere", "cohere",
"cohere2",
"minicpm", "minicpm",
"deepseek", "deepseek",
"olmo2", "olmo2",