Skip to content

Commit 7fd96a3

Browse files
Update examples/dreambooth/train_dreambooth_lora_sdxl.py
Co-authored-by: Linoy Tsaban <[email protected]>
1 parent 24c449c commit 7fd96a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/train_dreambooth_lora_sdxl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1212,7 +1212,7 @@ def main(args):
12121212
def get_lora_config(rank, dropout, use_dora, target_modules):
12131213
base_config = {
12141214
"r": rank,
1215-
"lora_alpha"=args.lora_alpha,
1215+
"lora_alpha":lora_alpha,
12161216
"lora_dropout": dropout,
12171217
"init_lora_weights": "gaussian",
12181218
"target_modules": target_modules,

0 commit comments

Comments
 (0)