removing redundant requires_grad = False (#10628)

We already set the unet to requires grad false at line 506 Co-authored-by: Aryan <[email protected]>
huggingface · Jan 23, 2025 · a451c0e · a451c0e
1 parent 37c9697
commit a451c0e
Showing 1 changed file with 0 additions and 4 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
@@ -515,10 +515,6 @@ def main():
     elif accelerator.mixed_precision == "bf16":
         weight_dtype = torch.bfloat16
 
-    # Freeze the unet parameters before adding adapters
-    for param in unet.parameters():
-        param.requires_grad_(False)
-
     unet_lora_config = LoraConfig(
         r=args.rank,
         lora_alpha=args.rank,