From 610a71d7d467e4cc892c824db882071ff0d282e1 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Tue, 2 Jul 2024 11:07:34 +0530 Subject: [PATCH] Fix indent in dreambooth lora advanced SD 15 script (#8753) update --- .../train_dreambooth_lora_sd15_advanced.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 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 index 02c7f8bb28..fea145d0b1 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py @@ -1856,10 +1856,10 @@ def main(args): generator = torch.Generator(device=accelerator.device).manual_seed(args.seed) if args.seed else None pipeline_args = {"prompt": args.validation_prompt} - if torch.backends.mps.is_available(): - autocast_ctx = nullcontext() - else: - autocast_ctx = torch.autocast(accelerator.device.type) + if torch.backends.mps.is_available(): + autocast_ctx = nullcontext() + else: + autocast_ctx = torch.autocast(accelerator.device.type) with autocast_ctx: images = [ @@ -1880,7 +1880,6 @@ def main(args): ] } ) - del pipeline torch.cuda.empty_cache()