diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/train_ti.py b/train_ti.py index 171d085..f78c7d2 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -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 | low_freq_noise=0, | ||
589 | strategy=textual_inversion_strategy, | 590 | strategy=textual_inversion_strategy, |
590 | num_train_epochs=args.num_train_epochs, | 591 | num_train_epochs=args.num_train_epochs, |
591 | sample_frequency=args.sample_frequency, | 592 | sample_frequency=args.sample_frequency, |