aboutsummaryrefslogtreecommitdiff
path: root/modules/paths.py
diff options
context:
space:
mode:
author不会画画的中医不是好程序员 <yfszzx@gmail.com>2022-10-10 20:21:25 +0800
committerGitHub <noreply@github.com>2022-10-10 20:21:25 +0800
commit1e18a5ffcc439b72adaaf425c0b79f3acb34322e (patch)
tree01f9c73c02076694a9bc3c965875646473771db8 /modules/paths.py
parent23f2989799ee3911d2959cfceb74b921f20c9a51 (diff)
parenta3578233395e585e68c2118d3630cb2a961d4a36 (diff)
Merge branch 'AUTOMATIC1111:master' into master
Diffstat (limited to 'modules/paths.py')
-rw-r--r--modules/paths.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/paths.py b/modules/paths.py
index 606f7d66..1e7a2fbc 100644
--- a/modules/paths.py
+++ b/modules/paths.py
@@ -1,6 +1,7 @@
import argparse
import os
import sys
+import modules.safe
script_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
models_path = os.path.join(script_path, "models")
@@ -12,6 +13,7 @@ possible_sd_paths = [os.path.join(script_path, 'repositories/stable-diffusion'),
for possible_sd_path in possible_sd_paths:
if os.path.exists(os.path.join(possible_sd_path, 'ldm/models/diffusion/ddpm.py')):
sd_path = os.path.abspath(possible_sd_path)
+ break
assert sd_path is not None, "Couldn't find Stable Diffusion in any of: " + str(possible_sd_paths)