diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-03 22:09:24 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-03 22:09:24 +0100 |
| commit | 220806dbd21da3ba83c14096225c31824dfe81df (patch) | |
| tree | a201272876bfe894f9d504d1582ac022add4b205 /train_ti.py | |
| parent | Implemented different noise offset (diff) | |
| download | textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.tar.gz textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.tar.bz2 textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.zip | |
Removed offset noise from training, added init offset to pipeline
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index bc9348d..c139cc0 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -607,7 +607,6 @@ def main(): | |||
| 607 | with_prior_preservation=args.num_class_images != 0, | 607 | with_prior_preservation=args.num_class_images != 0, |
| 608 | prior_loss_weight=args.prior_loss_weight, | 608 | prior_loss_weight=args.prior_loss_weight, |
| 609 | no_val=args.valid_set_size == 0, | 609 | no_val=args.valid_set_size == 0, |
| 610 | noise_offset=0, | ||
| 611 | strategy=textual_inversion_strategy, | 610 | strategy=textual_inversion_strategy, |
| 612 | num_train_epochs=args.num_train_epochs, | 611 | num_train_epochs=args.num_train_epochs, |
| 613 | gradient_accumulation_steps=args.gradient_accumulation_steps, | 612 | gradient_accumulation_steps=args.gradient_accumulation_steps, |