From cf002312e0b44cfde63172a629d5934cfacfaef6 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 26 Feb 2024 14:28:43 -0500 Subject: [PATCH] hotfix for lora rank (#1332) --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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