summaryrefslogtreecommitdiffstats
path: root/train_ti.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-04-02 11:14:03 +0200
committerVolpeon <git@volpeon.ink>2023-04-02 11:14:03 +0200
commite3669927b47b5367a3348d30c4b318da84af661d (patch)
treea9740db8cea9149eaab99f08f7cb8778f8e643b7 /train_ti.py
parentUpdate (diff)
downloadtextual-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.py2
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(