summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--.gitignore4
-rw-r--r--dreambooth.py2
-rw-r--r--infer.py2
-rw-r--r--textual_inversion.py2
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/
159# option (not recommended) you can uncomment the following to ignore the entire idea folder. 159# option (not recommended) you can uncomment the following to ignore the entire idea folder.
160#.idea/ 160#.idea/
161 161
162text-inversion-model/ 162output/
163dreambooth-model/
164inference/
165conf*.json 163conf*.json
166v1-inference.yaml* 164v1-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():
65 parser.add_argument( 65 parser.add_argument(
66 "--output_dir", 66 "--output_dir",
67 type=str, 67 type=str,
68 default="dreambooth-model", 68 default="output/dreambooth",
69 help="The output directory where the model predictions and checkpoints will be written.", 69 help="The output directory where the model predictions and checkpoints will be written.",
70 ) 70 )
71 parser.add_argument( 71 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():
57 parser.add_argument( 57 parser.add_argument(
58 "--output_dir", 58 "--output_dir",
59 type=str, 59 type=str,
60 default="inference", 60 default="output/inference",
61 ) 61 )
62 parser.add_argument( 62 parser.add_argument(
63 "--config", 63 "--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():
75 parser.add_argument( 75 parser.add_argument(
76 "--output_dir", 76 "--output_dir",
77 type=str, 77 type=str,
78 default="text-inversion-model", 78 default="output/text-inversion",
79 help="The output directory where the model predictions and checkpoints will be written.", 79 help="The output directory where the model predictions and checkpoints will be written.",
80 ) 80 )
81 parser.add_argument( 81 parser.add_argument(