Skip to content

Commit fab8c37

Browse files
committed
add use_dora check
1 parent 4e3e587 commit fab8c37

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,8 @@ def save_model_card(
9090
repo_folder=None,
9191
vae_path=None,
9292
):
93+
lora = "lora" if not use_dora else "dora"
94+
9395
widget_dict = []
9496
if images is not None:
9597
for i, image in enumerate(images):

0 commit comments

Comments
 (0)