add use_dora check

huggingface · Apr 17, 2024 · fab8c37 · fab8c37
1 parent 4e3e587
commit fab8c37
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
@@ -90,6 +90,8 @@ def save_model_card(
     repo_folder=None,
     vae_path=None,
 ):
+    lora = "lora" if not use_dora else "dora"
+
     widget_dict = []
     if images is not None:
         for i, image in enumerate(images):