From d673760fc671d665aadae3b032f8e99f21ab986d Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 16 Feb 2023 09:16:05 +0100
Subject: Integrated WIP UniPC scheduler

---
 train_ti.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'train_ti.py')

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():
     parser.add_argument(
         "--sample_steps",
         type=int,
-        default=20,
+        default=10,
         help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.",
     )
     parser.add_argument(
@@ -586,6 +586,7 @@ def main():
         seed=args.seed,
         with_prior_preservation=args.num_class_images != 0,
         prior_loss_weight=args.prior_loss_weight,
+        no_val=args.valid_set_size == 0,
         low_freq_noise=0,
         strategy=textual_inversion_strategy,
         num_train_epochs=args.num_train_epochs,
-- 
cgit v1.2.3-54-g00ecf