From f24db8802bd56948727a8c06b00ada367249766e Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 27 Sep 2022 22:25:43 +0200
Subject: Better output dir structure

---
 .gitignore           | 4 +---
 dreambooth.py        | 2 +-
 infer.py             | 2 +-
 textual_inversion.py | 2 +-
 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(
-- 
cgit v1.2.3-54-g00ecf