diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 4d7d8280e..44111b577 100644 --- 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