diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 09:08:45 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 09:08:45 +0000 |
| commit | 345028099d893f8a66726cfd13627d8cc1bcc724 (patch) | |
| tree | acb1da553620b7e7139db840ef43accf71b786a8 /modules/paths.py | |
| parent | d7b67d9b40e47ede766d3beb149b0c2b74651ece (diff) | |
| download | stable-diffusion-webui-gfx803-345028099d893f8a66726cfd13627d8cc1bcc724.tar.gz stable-diffusion-webui-gfx803-345028099d893f8a66726cfd13627d8cc1bcc724.tar.bz2 stable-diffusion-webui-gfx803-345028099d893f8a66726cfd13627d8cc1bcc724.zip | |
split codebase into multiple files; to anyone this affects negatively: sorry
Diffstat (limited to 'modules/paths.py')
| -rw-r--r-- | modules/paths.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/modules/paths.py b/modules/paths.py new file mode 100644 index 00000000..6d11b304 --- /dev/null +++ b/modules/paths.py @@ -0,0 +1,21 @@ +import argparse
+import os
+import sys
+
+script_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+sys.path.insert(0, script_path)
+
+# use current directory as SD dir if it has related files, otherwise parent dir of script as stated in guide
+sd_path = os.path.abspath('.') if os.path.exists('./ldm/models/diffusion/ddpm.py') else os.path.dirname(script_path)
+
+# add parent directory to path; this is where Stable diffusion repo should be
+path_dirs = [
+ (sd_path, 'ldm', 'Stable Diffusion'),
+ (os.path.join(sd_path, '../taming-transformers'), 'taming', 'Taming Transformers')
+]
+for d, must_exist, what in path_dirs:
+ must_exist_path = os.path.abspath(os.path.join(script_path, d, must_exist))
+ if not os.path.exists(must_exist_path):
+ print(f"Warning: {what} not found at path {must_exist_path}", file=sys.stderr)
+ else:
+ sys.path.append(os.path.join(script_path, d))
|