diff options
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" |