diff options
author | Volpeon <git@volpeon.ink> | 2023-02-16 09:16:05 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-02-16 09:16:05 +0100 |
commit | d673760fc671d665aadae3b032f8e99f21ab986d (patch) | |
tree | 7c14a998742b19ddecac6ee25a669892b41c305e /train_lora.py | |
parent | Update (diff) | |
download | textual-inversion-diff-d673760fc671d665aadae3b032f8e99f21ab986d.tar.gz textual-inversion-diff-d673760fc671d665aadae3b032f8e99f21ab986d.tar.bz2 textual-inversion-diff-d673760fc671d665aadae3b032f8e99f21ab986d.zip |
Integrated WIP UniPC scheduler
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index a8c1cf6..330bcd6 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -335,7 +335,7 @@ def parse_args(): | |||
335 | parser.add_argument( | 335 | parser.add_argument( |
336 | "--sample_steps", | 336 | "--sample_steps", |
337 | type=int, | 337 | type=int, |
338 | default=20, | 338 | default=10, |
339 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", | 339 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", |
340 | ) | 340 | ) |
341 | parser.add_argument( | 341 | parser.add_argument( |
@@ -487,6 +487,7 @@ def main(): | |||
487 | dtype=weight_dtype, | 487 | dtype=weight_dtype, |
488 | with_prior_preservation=args.num_class_images != 0, | 488 | with_prior_preservation=args.num_class_images != 0, |
489 | prior_loss_weight=args.prior_loss_weight, | 489 | prior_loss_weight=args.prior_loss_weight, |
490 | no_val=args.valid_set_size == 0, | ||
490 | ) | 491 | ) |
491 | 492 | ||
492 | checkpoint_output_dir = output_dir / "model" | 493 | checkpoint_output_dir = output_dir / "model" |