Unverified Commit 610a71d7 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix indent in dreambooth lora advanced SD 15 script (#8753)

update
parent c104482b
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment