added detection if SD repo is located in a current directory

1 files changed, 3 insertions, 1 deletions

diff --git a/webui.py b/webui.py
index 2b465396..49002a2a 100644
--- a/webui.py
+++ b/webui.py
@@ -3,7 +3,9 @@ import os
 import sys

 

 script_path = os.path.dirname(os.path.realpath(__file__))

-sd_path = os.path.dirname(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 = [