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 1be7df0 commit e83b436Copy full SHA for e83b436
examples/text_to_image/train_text_to_image_lora.py
@@ -418,9 +418,9 @@ def main():
418
# freeze parameters of models to save more memory
419
unet.requires_grad_(False)
420
vae.requires_grad_(False)
421
-
+
422
text_encoder.requires_grad_(False)
423
424
# For mixed precision training we cast the text_encoder and vae weights to half-precision
425
# as these models are only used for inference, keeping weights in full precision is not required.
426
weight_dtype = torch.float32
0 commit comments