lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-05 07:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

revise space

Browse Source
This commit is contained in:
layerdiffusion
2024-08-20 23:01:34 -07:00
parent 8b90bd88b5
commit 389d011fee
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modules_forge/forge_space.py
View File

@@ -161,8 +161,13 @@ class ForgeSpace:
original_cwd = os.getcwd()
os.chdir(self.hf_path)
if 'models' in sys.modules:
del sys.modules['models']
unsafe_module_prefixes = ['models', 'annotator']
modules_backup = {}
for module_name in list(sys.modules.keys()):
if any(module_name.startswith(prefix + '.') or module_name == prefix for prefix in unsafe_module_prefixes):
modules_backup[module_name] = sys.modules[module_name]
del sys.modules[module_name]
memory_management.unload_all_models()
sys.path.insert(0, self.hf_path)
@@ -187,8 +192,7 @@ class ForgeSpace:
server_port=port
)
if module_name in sys.modules:
del sys.modules[module_name]
sys.modules.update(modules_backup)
if 'models' in sys.modules:
del sys.modules['models']