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_dreambooth.py | 7 -------
 1 file changed, 7 deletions(-)

(limited to 'train_dreambooth.py')

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 0543a35..939a8f3 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -193,11 +193,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,
@@ -874,7 +869,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,
@@ -893,7 +887,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