We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b1c81f commit 8e0da14Copy full SHA for 8e0da14
examples/textual_inversion/textual_inversion.py
@@ -482,9 +482,9 @@ def main():
482
unet.enable_gradient_checkpointing()
483
484
weight_dtype = torch.float32
485
- if args.mixed_precision == 'fp16':
+ if args.mixed_precision == "fp16":
486
weight_dtype = torch.float16
487
- elif args.mixed_precision == 'bf16':
+ elif args.mixed_precision == "bf16":
488
weight_dtype = torch.bfloat16
489
490
# Move vae and unet to device.
0 commit comments