From e83b43612bb028052fe8038fc608e4f2442d0658 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 10 Feb 2023 13:07:46 +0200 Subject: [PATCH] make style --- examples/text_to_image/train_text_to_image_lora.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/text_to_image/train_text_to_image_lora.py b/examples/text_to_image/train_text_to_image_lora.py index a3c5bef73a..abc535594d 100644 --- a/examples/text_to_image/train_text_to_image_lora.py +++ b/examples/text_to_image/train_text_to_image_lora.py @@ -418,9 +418,9 @@ def main(): # freeze parameters of models to save more memory unet.requires_grad_(False) vae.requires_grad_(False) - + text_encoder.requires_grad_(False) - + # 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. weight_dtype = torch.float32