diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 2 |
1 files changed, 1 insertions, 1 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( |