4 files changed, 4 insertions, 6 deletions

diff --git a/.gitignore b/.gitignore
index 218c628..4456cef 100644
--- a/.gitignore
+++ b/.gitignore
@@ -159,8 +159,6 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
-text-inversion-model/
-dreambooth-model/
-inference/
+output/
 conf*.json
 v1-inference.yaml*
diff --git a/dreambooth.py b/dreambooth.py
index bc7a472..7c0d32c 100644
--- a/dreambooth.py
+++ b/dreambooth.py
@@ -65,7 +65,7 @@ def parse_args():
     parser.add_argument(
         "--output_dir",
         type=str,
-        default="dreambooth-model",
+        default="output/dreambooth",
         help="The output directory where the model predictions and checkpoints will be written.",
     )
     parser.add_argument(
diff --git a/infer.py b/infer.py
index b9e9ff7..70da08f 100644
--- a/infer.py
+++ b/infer.py
@@ -57,7 +57,7 @@ def parse_args():
     parser.add_argument(
         "--output_dir",
         type=str,
-        default="inference",
+        default="output/inference",
     )
     parser.add_argument(
         "--config",
diff --git a/textual_inversion.py b/textual_inversion.py
index aa8e744..81f1cf5 100644
--- a/textual_inversion.py
+++ b/textual_inversion.py
@@ -75,7 +75,7 @@ def parse_args():
     parser.add_argument(
         "--output_dir",
         type=str,
-        default="text-inversion-model",
+        default="output/text-inversion",
         help="The output directory where the model predictions and checkpoints will be written.",
     )
     parser.add_argument(