From 3acc0e0eeb32e20beedf8f914134e145d5eccdef Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Fri, 8 Dec 2023 11:45:27 +0000 Subject: [PATCH] [Advanced Diffusion Script] Add Widget default text (#6100) add widget --- .../train_dreambooth_lora_sdxl_advanced.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py index f4b4e42c8b19d..4d4ec523bfe6d 100644 --- 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}' --- """