From c55920681ccd15c183f168619d9b332cd837163c Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 3 Jan 2023 13:32:37 +0100
Subject: Don't use vector_dropout by default

---
 train_dreambooth.py | 2 +-
 train_ti.py         | 2 +-
 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(
-- 
cgit v1.2.3-70-g09d2