Update train.py

ExponentialML · web-flow · commit fc6b82753b2b · 2023-04-09T02:07:10.000-07:00
diff --git a/train.py b/train.py
@@ -713,7 +713,7 @@ def finetune_unet(batch, train_encoder=False):
         if text_trainable:
             text_encoder.train()
 
-            if use_text_lora and text_encoder_gradient_checkpointing: 
+            if use_text_lora: 
                 text_encoder.text_model.embeddings.requires_grad_(True)
 
             if global_step == 0 and train_text_encoder:
