StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #10025 from acncagua/Upscaler_initialization

Browse Source 
Initialize the upscalers
This commit is contained in:
AUTOMATIC1111 2023-05-08 15:03:59 +03:00 committed by GitHub
commit b3a44385b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/modelloader.py
View File

@ -122,12 +122,9 @@ forbidden_upscaler_classes = set()
def list_builtin_upscalers(): def list_builtin_upscalers():
load_upscalers()
builtin_upscaler_classes.clear() builtin_upscaler_classes.clear()
builtin_upscaler_classes.extend(Upscaler.__subclasses__()) builtin_upscaler_classes.extend(Upscaler.__subclasses__())
def forbid_loaded_nonbuiltin_upscalers(): def forbid_loaded_nonbuiltin_upscalers():
for cls in Upscaler.__subclasses__(): for cls in Upscaler.__subclasses__():
if cls not in builtin_upscaler_classes: if cls not in builtin_upscaler_classes:

3
webui.py
View File

@ -187,6 +187,9 @@ def initialize():
modules.scripts.load_scripts() modules.scripts.load_scripts()
startup_timer.record("load scripts") startup_timer.record("load scripts")
modelloader.load_upscalers()
#startup_timer.record("load upscalers") #Is this necessary? I don't know.
modules.sd_vae.refresh_vae_list() modules.sd_vae.refresh_vae_list()
startup_timer.record("refresh VAE") startup_timer.record("refresh VAE")