zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-25 18:07:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: check LoRA layers number error (#446)

Browse Source
This commit is contained in:
Madroid Ma 2024-02-16 22:03:33 +08:00 committed by GitHub
parent 837a02092d
commit 726b1ddec0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -21,7 +21,7 @@ def linear_to_lora_layers(model: nn.Module, num_lora_layers: int):
if num_lora_layers > num_model: if num_lora_layers > num_model:
raise ValueError( raise ValueError(
f"Requested {num_lora_layers} LoRA layers " f"Requested {num_lora_layers} LoRA layers "
f"but the model only has {num_model_layers} layers." f"but the model only has {num_model} layers."
) )
if model.model_type in [ if model.model_type in [