Integrate Self-Attention-Guided (SAG) Stable Diffusion in my custom pipeline

1 files changed, 0 insertions, 8 deletions

diff --git a/train_lora.py b/train_lora.py
index ab1753b..5fd05cc 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -178,11 +178,6 @@ def parse_args():
         help="Number of updates steps to accumulate before performing a backward/update pass.",
     )
     parser.add_argument(
-        "--gradient_checkpointing",
-        action="store_true",
-        help="Whether or not to use gradient checkpointing to save memory at the expense of slower backward pass.",
-    )
-    parser.add_argument(
         "--find_lr",
         action="store_true",
         help="Automatically find a learning rate (no training).",
@@ -429,9 +424,6 @@ def main():
     vae.set_use_memory_efficient_attention_xformers(True)
     unet.enable_xformers_memory_efficient_attention()
 
-    if args.gradient_checkpointing:
-        unet.enable_gradient_checkpointing()
-
     unet.to(accelerator.device, dtype=weight_dtype)
     text_encoder.to(accelerator.device, dtype=weight_dtype)