lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-06 13:39:51 +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:53:49 -08:00
parent 4b82188b6e
commit 20c785705f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -24,7 +24,9 @@ for k in ui_preprocessor_keys:
p = LegacyPreprocessor()
p.name = k
real_key = reverse_preprocessor_aliases.get(k, k)
assert real_key in cn_preprocessor_modules
if real_key not in cn_preprocessor_modules:
print(f'bad key {real_key}')
continue
p.legacy_call_function = cn_preprocessor_modules[real_key]
p.legacy_unload_function = cn_preprocessor_unloadable.get(real_key, None)
legacy_preprocessors[k] = p