summaryrefslogtreecommitdiffstats
path: root/train_dreambooth.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r--train_dreambooth.py7
1 files changed, 0 insertions, 7 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py
index 0543a35..939a8f3 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -194,11 +194,6 @@ def parse_args():
194 help="Shuffle tags.", 194 help="Shuffle tags.",
195 ) 195 )
196 parser.add_argument( 196 parser.add_argument(
197 "--guidance_scale",
198 type=float,
199 default=0,
200 )
201 parser.add_argument(
202 "--num_class_images", 197 "--num_class_images",
203 type=int, 198 type=int,
204 default=0, 199 default=0,
@@ -874,7 +869,6 @@ def main():
874 dtype=weight_dtype, 869 dtype=weight_dtype,
875 seed=args.seed, 870 seed=args.seed,
876 compile_unet=args.compile_unet, 871 compile_unet=args.compile_unet,
877 guidance_scale=args.guidance_scale,
878 prior_loss_weight=args.prior_loss_weight if args.num_class_images != 0 else 0, 872 prior_loss_weight=args.prior_loss_weight if args.num_class_images != 0 else 0,
879 sample_scheduler=sample_scheduler, 873 sample_scheduler=sample_scheduler,
880 sample_batch_size=args.sample_batch_size, 874 sample_batch_size=args.sample_batch_size,
@@ -893,7 +887,6 @@ def main():
893 tokenizer=tokenizer, 887 tokenizer=tokenizer,
894 constant_prompt_length=args.compile_unet, 888 constant_prompt_length=args.compile_unet,
895 class_subdir=args.class_image_dir, 889 class_subdir=args.class_image_dir,
896 with_guidance=args.guidance_scale != 0,
897 num_class_images=args.num_class_images, 890 num_class_images=args.num_class_images,
898 size=args.resolution, 891 size=args.resolution,
899 num_buckets=args.num_buckets, 892 num_buckets=args.num_buckets,