hotfix for lora rank (#1332)

2024-02-26 14:28:43 -05:00
parent 7de912e097
commit cf002312e0
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/config/models/input/v0_4_1/init.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/init.py
@@ -146,7 +146,7 @@ class LoraConfig(BaseModel):

    adapter: Optional[str] = None
    lora_model_dir: Optional[str] = None
-    lora_rank: Optional[int] = None
+    lora_r: Optional[int] = None
    lora_alpha: Optional[int] = None
    lora_fan_in_fan_out: Optional[bool] = None
    lora_target_modules: Optional[List[str]] = None