diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 7 |
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, |