diff options
| author | Volpeon <git@volpeon.ink> | 2022-09-27 22:25:43 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-09-27 22:25:43 +0200 |
| commit | f24db8802bd56948727a8c06b00ada367249766e (patch) | |
| tree | b4710f9ccb4a981d00ebad95c1a63be93fa32de4 | |
| parent | Various improvements, added inference script (diff) | |
| download | textual-inversion-diff-f24db8802bd56948727a8c06b00ada367249766e.tar.gz textual-inversion-diff-f24db8802bd56948727a8c06b00ada367249766e.tar.bz2 textual-inversion-diff-f24db8802bd56948727a8c06b00ada367249766e.zip | |
Better output dir structure
| -rw-r--r-- | .gitignore | 4 | ||||
| -rw-r--r-- | dreambooth.py | 2 | ||||
| -rw-r--r-- | infer.py | 2 | ||||
| -rw-r--r-- | textual_inversion.py | 2 |
4 files changed, 4 insertions, 6 deletions
| @@ -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 | ||
| 162 | text-inversion-model/ | 162 | output/ |
| 163 | dreambooth-model/ | ||
| 164 | inference/ | ||
| 165 | conf*.json | 163 | conf*.json |
| 166 | v1-inference.yaml* | 164 | 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(): | |||
| 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( |
| @@ -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( |