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_ti.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_ti.py')
-rw-r--r-- | train_ti.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index f78c7d2..d1defb3 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -392,7 +392,7 @@ def parse_args(): | |||
392 | parser.add_argument( | 392 | parser.add_argument( |
393 | "--sample_steps", | 393 | "--sample_steps", |
394 | type=int, | 394 | type=int, |
395 | default=20, | 395 | default=10, |
396 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", | 396 | help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", |
397 | ) | 397 | ) |
398 | parser.add_argument( | 398 | parser.add_argument( |
@@ -586,6 +586,7 @@ def main(): | |||
586 | seed=args.seed, | 586 | seed=args.seed, |
587 | with_prior_preservation=args.num_class_images != 0, | 587 | with_prior_preservation=args.num_class_images != 0, |
588 | prior_loss_weight=args.prior_loss_weight, | 588 | prior_loss_weight=args.prior_loss_weight, |
589 | no_val=args.valid_set_size == 0, | ||
589 | low_freq_noise=0, | 590 | low_freq_noise=0, |
590 | strategy=textual_inversion_strategy, | 591 | strategy=textual_inversion_strategy, |
591 | num_train_epochs=args.num_train_epochs, | 592 | num_train_epochs=args.num_train_epochs, |