use correct tags for lora

huggingface · Jun 10, 2024 · 89b945d · 89b945d
1 parent fab8c37
commit 89b945d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
@@ -199,7 +199,7 @@ def save_model_card(
     tags = ["text-to-image", 
             "diffusers", 
             "diffusers-training",
-            "lora",
+            lora,
             "template:sd-lora"
             "stable-diffusion", 
             "stable-diffusion-diffusers"]