diff options
| author | Volpeon <git@volpeon.ink> | 2023-06-22 18:34:53 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-06-22 18:34:53 +0200 |
| commit | 01f0b3bd5a7965776b420c97056f82601e2b7312 (patch) | |
| tree | 236bde99b6116abdc7ca75fb57828ce21b74ba32 /train_ti.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.tar.gz textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.tar.bz2 textual-inversion-diff-01f0b3bd5a7965776b420c97056f82601e2b7312.zip | |
Added prompt dropout
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 7f93960..1dbd637 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -923,7 +923,7 @@ def main(): | |||
| 923 | progressive_buckets=args.progressive_buckets, | 923 | progressive_buckets=args.progressive_buckets, |
| 924 | bucket_step_size=args.bucket_step_size, | 924 | bucket_step_size=args.bucket_step_size, |
| 925 | bucket_max_pixels=args.bucket_max_pixels, | 925 | bucket_max_pixels=args.bucket_max_pixels, |
| 926 | dropout=args.tag_dropout, | 926 | tag_dropout=args.tag_dropout, |
| 927 | shuffle=not args.no_tag_shuffle, | 927 | shuffle=not args.no_tag_shuffle, |
| 928 | template_key=data_template, | 928 | template_key=data_template, |
| 929 | placeholder_tokens=args.placeholder_tokens, | 929 | placeholder_tokens=args.placeholder_tokens, |