36 lines
913 B
Python
36 lines
913 B
Python
import os
|
|
import sys
|
|
|
|
debug = False
|
|
is_auto = False
|
|
try:
|
|
from modules import shared
|
|
is_auto = True
|
|
except:
|
|
pass
|
|
|
|
if not is_auto:
|
|
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", ".."))
|
|
if debug:
|
|
print(f"Base dir (sdplus) is: {base_dir} from {__file__}")
|
|
ext_dir = os.path.join(base_dir, 'core', 'modules', 'dreambooth')
|
|
else:
|
|
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
|
if debug:
|
|
print(f"Base dir (auto1111) is: {base_dir} from {__file__}")
|
|
ext_dir = os.path.join(base_dir, 'extensions', 'sd_dreambooth_extension')
|
|
|
|
if ext_dir not in sys.path:
|
|
if debug:
|
|
print(f"Appending (install): {ext_dir}")
|
|
sys.path.insert(0, ext_dir)
|
|
else:
|
|
if debug:
|
|
print(f"Ext dir already in path? {ext_dir}")
|
|
print(sys.path)
|
|
|
|
|
|
from postinstall import actual_install
|
|
|
|
actual_install()
|