diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-26 14:27:54 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-26 14:27:54 +0200 |
| commit | 19ae465203c8dcc0b1179584db632015362b5e44 (patch) | |
| tree | ad6d45e78826f525c336927e4269197667f1f354 /train_dreambooth.py | |
| parent | Fix training with guidance (diff) | |
| download | textual-inversion-diff-19ae465203c8dcc0b1179584db632015362b5e44.tar.gz textual-inversion-diff-19ae465203c8dcc0b1179584db632015362b5e44.tar.bz2 textual-inversion-diff-19ae465203c8dcc0b1179584db632015362b5e44.zip | |
Improved inverted tokens
Diffstat (limited to 'train_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index 7a33bca..9345797 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -585,6 +585,7 @@ def main(): | |||
| 585 | dropout=args.tag_dropout, | 585 | dropout=args.tag_dropout, |
| 586 | shuffle=not args.no_tag_shuffle, | 586 | shuffle=not args.no_tag_shuffle, |
| 587 | template_key=args.train_data_template, | 587 | template_key=args.train_data_template, |
| 588 | placeholder_tokens=args.placeholder_tokens, | ||
| 588 | valid_set_size=args.valid_set_size, | 589 | valid_set_size=args.valid_set_size, |
| 589 | train_set_pad=args.train_set_pad, | 590 | train_set_pad=args.train_set_pad, |
| 590 | valid_set_pad=args.valid_set_pad, | 591 | valid_set_pad=args.valid_set_pad, |