Integrated WIP UniPC scheduler

1 files changed, 2 insertions, 1 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 4c1ec31..5a7911c 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -375,7 +375,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(
@@ -511,6 +511,7 @@ def main():
         dtype=weight_dtype,
         with_prior_preservation=args.num_class_images != 0,
         prior_loss_weight=args.prior_loss_weight,
+        no_val=args.valid_set_size == 0,
     )
 
     checkpoint_output_dir = output_dir / "model"