added detection if SD repo is located in a current directory
This commit is contained in:
parent
b90feb02aa
commit
d5dfbc45a6
4
webui.py
4
webui.py
|
@ -3,7 +3,9 @@ import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
script_path = os.path.dirname(os.path.realpath(__file__))
|
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
|
# add parent directory to path; this is where Stable diffusion repo should be
|
||||||
path_dirs = [
|
path_dirs = [
|
||||||
|
|
Loading…
Reference in New Issue