diff options
author | Volpeon <git@volpeon.ink> | 2023-04-02 11:14:03 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-04-02 11:14:03 +0200 |
commit | e3669927b47b5367a3348d30c4b318da84af661d (patch) | |
tree | a9740db8cea9149eaab99f08f7cb8778f8e643b7 /train_ti.py | |
parent | Update (diff) | |
download | textual-inversion-diff-e3669927b47b5367a3348d30c4b318da84af661d.tar.gz textual-inversion-diff-e3669927b47b5367a3348d30c4b318da84af661d.tar.bz2 textual-inversion-diff-e3669927b47b5367a3348d30c4b318da84af661d.zip |
Update dataset format: Separate prompt and keywords
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 5482326..651dfbe 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -186,7 +186,7 @@ def parse_args(): | |||
186 | parser.add_argument( | 186 | parser.add_argument( |
187 | "--tag_dropout", | 187 | "--tag_dropout", |
188 | type=float, | 188 | type=float, |
189 | default=0, | 189 | default=0.1, |
190 | help="Tag dropout probability.", | 190 | help="Tag dropout probability.", |
191 | ) | 191 | ) |
192 | parser.add_argument( | 192 | parser.add_argument( |