Skip to content

GitLab

Commit e83b4361 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 1be7df02
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/text_to_image/train_text_to_image_lora.py
View file @ e83b4361
...@@ -418,9 +418,9 @@ def main(): ...@@ -418,9 +418,9 @@ def main():
# freeze parameters of models to save more memory # freeze parameters of models to save more memory
unet.requires_grad_(False) unet.requires_grad_(False)
vae.requires_grad_(False) vae.requires_grad_(False)
text_encoder.requires_grad_(False) text_encoder.requires_grad_(False)
# For mixed precision training we cast the text_encoder and vae weights to half-precision # For mixed precision training we cast the text_encoder and vae weights to half-precision
# as these models are only used for inference, keeping weights in full precision is not required. # as these models are only used for inference, keeping weights in full precision is not required.
weight_dtype = torch.float32 weight_dtype = torch.float32
......
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