diff options
author | Volpeon <git@volpeon.ink> | 2023-06-22 07:34:04 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-06-22 07:34:04 +0200 |
commit | 11b7740deeef7903e81ba4c65a45853323a5fd5e (patch) | |
tree | 0d5eec0e574447afe80538252884e0093742e3cb /train_dreambooth.py | |
parent | Remove convnext (diff) | |
download | textual-inversion-diff-11b7740deeef7903e81ba4c65a45853323a5fd5e.tar.gz textual-inversion-diff-11b7740deeef7903e81ba4c65a45853323a5fd5e.tar.bz2 textual-inversion-diff-11b7740deeef7903e81ba4c65a45853323a5fd5e.zip |
Remove training guidance_scale
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, |