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():
     parser.add_argument(
         "--vector_dropout",
         type=int,
-        default=0.1,
+        default=0,
         help="Vector dropout probability.",
     )
     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():
     parser.add_argument(
         "--vector_dropout",
         type=int,
-        default=0.1,
+        default=0,
         help="Vector dropout probability.",
     )
     parser.add_argument(