Skip to content

Commit 8e0da14

Browse files
committed
Fix formatting
1 parent 7b1c81f commit 8e0da14

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/textual_inversion/textual_inversion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -482,9 +482,9 @@ def main():
482482
unet.enable_gradient_checkpointing()
483483

484484
weight_dtype = torch.float32
485-
if args.mixed_precision == 'fp16':
485+
if args.mixed_precision == "fp16":
486486
weight_dtype = torch.float16
487-
elif args.mixed_precision == 'bf16':
487+
elif args.mixed_precision == "bf16":
488488
weight_dtype = torch.bfloat16
489489

490490
# Move vae and unet to device.

0 commit comments

Comments
 (0)