[Advanced Diffusion Script] Add Widget default text (huggingface#6100)

add widget
AmericanPresidentJimmyCarter · Apr 26, 2024 · 3acc0e0 · 3acc0e0
1 parent 36a587a
commit 3acc0e0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
@@ -157,6 +157,8 @@ def save_model_card(
 base_model: {base_model}
 instance_prompt: {instance_prompt}
 license: openrail++
+widget:
+    - text: '{validation_prompt if validation_prompt else instance_prompt}'
 ---
 """