lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-11 07:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update legacy_preprocessor_loader.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 15:51:44 -08:00
parent c2d53721e4
commit 4b82188b6e
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessor_loader.py
View File

@@ -7,23 +7,26 @@ from modules_forge.shared import Preprocessor, PreprocessorParameter, preprocess
# how to make better implementation of preprocessors.
# No newer preprocessors should be written in this legacy way.
from legacy_preprocessors.preprocessor_meta import cn_preprocessor_modules, cn_preprocessor_unloadable
from legacy_preprocessors.preprocessor_meta import cn_preprocessor_modules, cn_preprocessor_unloadable, ui_preprocessor_keys, reverse_preprocessor_aliases
class LegacyPreprocessor(Preprocessor):
def __init__(self):
super().__init__()
legacy_call_function = None
legacy_unload_function = None
self.legacy_call_function = None
self.legacy_unload_function = None
return
legacy_preprocessors = {}
for k, v in cn_preprocessor_modules.items():
for k in ui_preprocessor_keys:
p = LegacyPreprocessor()
p.name = k
p.legacy_call_function = v
real_key = reverse_preprocessor_aliases.get(k, k)
assert real_key in cn_preprocessor_modules
p.legacy_call_function = cn_preprocessor_modules[real_key]
p.legacy_unload_function = cn_preprocessor_unloadable.get(real_key, None)
legacy_preprocessors[k] = p