diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-15 15:11:37 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-15 15:11:37 +0100 |
| commit | 0b8568ebfb484b9155d19c567e890ff52776544d (patch) | |
| tree | 09ff3256ca8d2c9359a236e7bdb274f539c5253e /train_ti.py | |
| parent | Improved batch padding (diff) | |
| download | textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.tar.gz textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.tar.bz2 textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.zip | |
Update
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, |