Remove training guidance_scale

1 files changed, 0 insertions, 7 deletions

diff --git a/train_lora.py b/train_lora.py
index b7ee2d6..51dc827 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -207,11 +207,6 @@ def parse_args():
         help="Shuffle tags.",
     )
     parser.add_argument(
-        "--guidance_scale",
-        type=float,
-        default=0,
-    )
-    parser.add_argument(
         "--num_class_images",
         type=int,
         default=0,
@@ -998,7 +993,6 @@ def main():
         dtype=weight_dtype,
         seed=args.seed,
         compile_unet=args.compile_unet,
-        guidance_scale=args.guidance_scale,
         prior_loss_weight=args.prior_loss_weight if args.num_class_images != 0 else 0,
         sample_scheduler=sample_scheduler,
         sample_batch_size=args.sample_batch_size,
@@ -1022,7 +1016,6 @@ def main():
         tokenizer=tokenizer,
         constant_prompt_length=args.compile_unet,
         class_subdir=args.class_image_dir,
-        with_guidance=args.guidance_scale != 0,
         num_class_images=args.num_class_images,
         size=args.resolution,
         num_buckets=args.num_buckets,