Skip to content

Commit 2b84334

Browse files
committed
fix use_dora with peft version
1 parent 936cbeb commit 2b84334

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

examples/dreambooth/train_dreambooth_lora_sdxl.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1191,12 +1191,13 @@ def get_lora_config(rank, use_dora, target_modules):
11911191
"init_lora_weights": "gaussian",
11921192
"target_modules": target_modules,
11931193
}
1194-
if use_dora and is_peft_version("<", "0.9.0"):
1195-
raise ValueError(
1196-
"You need `peft` 0.9.0 at least to use DoRA-enabled LoRAs. Please upgrade your installation of `peft`."
1197-
)
1198-
else:
1199-
base_config["use_dora"] = True
1194+
if use_dora:
1195+
if is_peft_version("<", "0.9.0"):
1196+
raise ValueError(
1197+
"You need `peft` 0.9.0 at least to use DoRA-enabled LoRAs. Please upgrade your installation of `peft`."
1198+
)
1199+
else:
1200+
base_config["use_dora"] = True
12001201

12011202
return LoraConfig(**base_config)
12021203

0 commit comments

Comments
 (0)