mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-01-01 12:25:06 +08:00
feat: ensure the indexability of dynamically imported packages
Signed-off-by: storyicon <storyicon@foxmail.com>
This commit is contained in:
parent
bef51aed03
commit
e73a7e4006
@ -2,13 +2,18 @@ import os
|
||||
import importlib.util
|
||||
|
||||
from modules import errors
|
||||
|
||||
import sys
|
||||
|
||||
def load_module(path):
|
||||
module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)
|
||||
module = importlib.util.module_from_spec(module_spec)
|
||||
module_spec.loader.exec_module(module)
|
||||
|
||||
if os.path.isfile(path):
|
||||
sp = os.path.splitext(path)
|
||||
module_name = sp[0]
|
||||
else:
|
||||
module_name = os.path.basename(path)
|
||||
sys.modules[module_name] = module
|
||||
return module
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user