fix: check LoRA layers number error (#446)

2024-02-16 22:03:33 +08:00 · 2024-02-16 22:03:33 +08:00 · 726b1ddec0
commit 726b1ddec0
parent 837a02092d
1 changed files with 1 additions and 1 deletions
--- a/llms/mlx_lm/tuner/utils.py
+++ b/llms/mlx_lm/tuner/utils.py
@ -21,7 +21,7 @@ def linear_to_lora_layers(model: nn.Module, num_lora_layers: int):
        if num_lora_layers > num_model:
            raise ValueError(
                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 [