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 c42338c0834e..b5336456ba87 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py @@ -2041,7 +2041,7 @@ def compute_text_embeddings(prompt, text_encoders, tokenizers): } ) - + if args.train_text_encoder_ti: embedding_handler.save_embeddings( f"{args.output_dir}/embeddings.safetensors",