Skip to content

Commit e83b436

Browse files
make style
1 parent 1be7df0 commit e83b436

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/text_to_image/train_text_to_image_lora.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -418,9 +418,9 @@ def main():
418418
# freeze parameters of models to save more memory
419419
unet.requires_grad_(False)
420420
vae.requires_grad_(False)
421-
421+
422422
text_encoder.requires_grad_(False)
423-
423+
424424
# For mixed precision training we cast the text_encoder and vae weights to half-precision
425425
# as these models are only used for inference, keeping weights in full precision is not required.
426426
weight_dtype = torch.float32

0 commit comments

Comments
 (0)