From 11b7740deeef7903e81ba4c65a45853323a5fd5e Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 22 Jun 2023 07:34:04 +0200
Subject: Remove training guidance_scale

---
 train_lora.py | 7 -------
 1 file changed, 7 deletions(-)

(limited to 'train_lora.py')

diff --git a/train_lora.py b/train_lora.py
index b7ee2d6..51dc827 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -206,11 +206,6 @@ def parse_args():
         action="store_true",
         help="Shuffle tags.",
     )
-    parser.add_argument(
-        "--guidance_scale",
-        type=float,
-        default=0,
-    )
     parser.add_argument(
         "--num_class_images",
         type=int,
@@ -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,
-- 
cgit v1.2.3-54-g00ecf