Fallback for extensions dir

Fixes #145
This commit is contained in:
Dominik Reh
2023-03-26 16:32:20 +02:00
parent afe3f23afa
commit 32c7749a5f

View File

@@ -4,14 +4,20 @@
import gradio as gr
from pathlib import Path
from modules import scripts, script_callbacks, shared, sd_hijack
from modules.paths import script_path, extensions_dir
import yaml
# Webui root path
FILE_DIR = Path(script_path)
try:
from modules.paths import script_path, extensions_dir
# Webui root path
FILE_DIR = Path(script_path)
# The extension base path
EXT_PATH = Path(extensions_dir)
# The extension base path
EXT_PATH = Path(extensions_dir)
except ImportError:
# Webui root path
FILE_DIR = Path().absolute()
# The extension base path
EXT_PATH = FILE_DIR.joinpath('extensions')
# Tags base path
TAGS_PATH = Path(scripts.basedir()).joinpath('tags')