Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
stable-diffusion-webui/modules/script_loading.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

39 lines
1.2 KiB
Python
Raw Normal View History

make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
import os
use python importlib to load and execute extension modules previously module attributes like __file__ where not set correctly, leading to scripts getting the directory of the stable-diffusion repo location instead of their own script. This causes problem when loading user data from an external location using the --data-dir flag, as extensions would look for their own code in the stable-diffusion repo location instead of the data dir location. Using pythons importlib functions sets the modules specs correctly and executes them. But this will break extensions if they build paths based on the previously incorrect __file__ attribute.
2023-01-25 09:55:59 -07:00
import importlib.util
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
rename print_error to report, use it with together with package name
2023-05-31 10:56:37 -06:00
from modules import errors
feat: ensure the indexability of dynamically imported packages Signed-off-by: storyicon <storyicon@foxmail.com>
2024-04-01 03:13:07 -06:00
import sys
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
use 'scripts.' prefix for names of dynamically loaded modules
2024-04-06 00:05:04 -06:00
loaded_scripts = {}
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
def load_module(path):
Fix cls.__module__
2024-04-15 20:14:19 -06:00
module_name, _ = os.path.splitext(os.path.basename(path))
full_module_name = "scripts." + module_name
module_spec = importlib.util.spec_from_file_location(full_module_name, path)
use python importlib to load and execute extension modules previously module attributes like __file__ where not set correctly, leading to scripts getting the directory of the stable-diffusion repo location instead of their own script. This causes problem when loading user data from an external location using the --data-dir flag, as extensions would look for their own code in the stable-diffusion repo location instead of the data dir location. Using pythons importlib functions sets the modules specs correctly and executes them. But this will break extensions if they build paths based on the previously incorrect __file__ attribute.
2023-01-25 09:55:59 -07:00
module = importlib.util.module_from_spec(module_spec)
module_spec.loader.exec_module(module)
Fix cls.__module__
2024-04-15 20:14:19 -06:00
loaded_scripts[full_module_name] = module
sys.modules[full_module_name] = module
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
return module
add backwards compatibility --lyco-dir-backcompat option, use that for LyCORIS directory instead of hardcoded value prevent running preload.py for disabled extensions
2023-07-18 11:11:30 -06:00
def preload_extensions(extensions_dir, parser, extension_list=None):
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
if not os.path.isdir(extensions_dir):
return
add backwards compatibility --lyco-dir-backcompat option, use that for LyCORIS directory instead of hardcoded value prevent running preload.py for disabled extensions
2023-07-18 11:11:30 -06:00
extensions = extension_list if extension_list is not None else os.listdir(extensions_dir)
for dirname in sorted(extensions):
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py
2022-11-12 00:56:06 -07:00
preload_script = os.path.join(extensions_dir, dirname, "preload.py")
if not os.path.isfile(preload_script):
continue
try:
module = load_module(preload_script)
if hasattr(module, 'preload'):
module.preload(parser)
except Exception:
rename print_error to report, use it with together with package name
2023-05-31 10:56:37 -06:00
errors.report(f"Error running preload() for {preload_script}", exc_info=True)