diff options
| author | Max Audron <audron@cocaine.farm> | 2023-01-25 16:15:42 +0000 |
|---|---|---|
| committer | Max Audron <audron@cocaine.farm> | 2023-01-27 13:44:30 +0000 |
| commit | 5eee2ac39863f9e44591b50d0710dd2615416a13 (patch) | |
| tree | 72d90317db4e724a9d9ace4f9e8d8f2d4467d5e9 /modules/textual_inversion/preprocess.py | |
| parent | 9beb794e0b0dc1a0f9e89d8e38bd789a8c608397 (diff) | |
| download | stable-diffusion-webui-gfx803-5eee2ac39863f9e44591b50d0710dd2615416a13.tar.gz stable-diffusion-webui-gfx803-5eee2ac39863f9e44591b50d0710dd2615416a13.tar.bz2 stable-diffusion-webui-gfx803-5eee2ac39863f9e44591b50d0710dd2615416a13.zip | |
add data-dir flag and set all user data directories based on it
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py index c0ac11d3..2239cb84 100644 --- a/modules/textual_inversion/preprocess.py +++ b/modules/textual_inversion/preprocess.py @@ -6,8 +6,7 @@ import sys import tqdm
import time
-from modules import shared, images, deepbooru
-from modules.paths import models_path
+from modules import paths, shared, images, deepbooru
from modules.shared import opts, cmd_opts
from modules.textual_inversion import autocrop
@@ -199,7 +198,7 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre dnn_model_path = None
try:
- dnn_model_path = autocrop.download_and_cache_models(os.path.join(models_path, "opencv"))
+ dnn_model_path = autocrop.download_and_cache_models(os.path.join(paths.models_path, "opencv"))
except Exception as e:
print("Unable to load face detection model for auto crop selection. Falling back to lower quality haar method.", e)
|