summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-03 13:32:37 +0100
committerVolpeon <git@volpeon.ink>2023-01-03 13:32:37 +0100
commitc55920681ccd15c183f168619d9b332cd837163c (patch)
tree9b86ba07b89a282153276577174820bcdb3f0387
parentAdded vector dropout (diff)
downloadtextual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.tar.gz
textual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.tar.bz2
textual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.zip
Don't use vector_dropout by default
-rw-r--r--train_dreambooth.py2
-rw-r--r--train_ti.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py
index f26b7f5..df8b54c 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -110,7 +110,7 @@ def parse_args():
110 parser.add_argument( 110 parser.add_argument(
111 "--vector_dropout", 111 "--vector_dropout",
112 type=int, 112 type=int,
113 default=0.1, 113 default=0,
114 help="Vector dropout probability.", 114 help="Vector dropout probability.",
115 ) 115 )
116 parser.add_argument( 116 parser.add_argument(
diff --git a/train_ti.py b/train_ti.py
index cacbbc7..1685dc4 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -157,7 +157,7 @@ def parse_args():
157 parser.add_argument( 157 parser.add_argument(
158 "--vector_dropout", 158 "--vector_dropout",
159 type=int, 159 type=int,
160 default=0.1, 160 default=0,
161 help="Vector dropout probability.", 161 help="Vector dropout probability.",
162 ) 162 )
163 parser.add_argument( 163 parser.add_argument(