soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

added detection if SD repo is located in a current directory

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-02 01:06:11 +03:00
parent b90feb02aa
commit d5dfbc45a6
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
webui.py
View File

@ -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 = [