Skip to content

Commit e988c7e

Browse files
committed
add use_dora check
1 parent f51f71a commit e988c7e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,8 @@ def save_model_card(
111111
repo_folder=None,
112112
vae_path=None,
113113
):
114+
lora = "lora" if not use_dora else "dora"
115+
114116
widget_dict = []
115117
if images is not None:
116118
for i, image in enumerate(images):

0 commit comments

Comments
 (0)