1 files changed, 0 insertions, 7 deletions

diff --git a/train_ti.py b/train_ti.py
index 7d1ef19..7f93960 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -125,11 +125,6 @@ def parse_args():
         action="store_true",
     )
     parser.add_argument(
-        "--guidance_scale",
-        type=float,
-        default=0,
-    )
-    parser.add_argument(
         "--num_class_images",
         type=int,
         default=0,
@@ -852,7 +847,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,
         no_val=args.valid_set_size == 0,
         strategy=textual_inversion_strategy,
@@ -923,7 +917,6 @@ def main():
             batch_size=args.train_batch_size,
             tokenizer=tokenizer,
             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,