aboutsummaryrefslogtreecommitdiffstats
path: root/webui.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-24 09:39:46 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-10-24 09:39:46 +0300
commit876a96f0f9843382ebc8984db3de5d8af0e9ce4c (patch)
tree8771f3f3f9918166be53dbe5d5d87c6ad23b409b /webui.py
parent999929bea4e66384e6cf8da1e2cf833d1fa41053 (diff)
downloadstable-diffusion-webui-gfx803-876a96f0f9843382ebc8984db3de5d8af0e9ce4c.tar.gz
remove erroneous dir in the extension directory
remove loading .js files from scripts dir (they go into javascript) load scripts after models, for scripts that depend on loaded models
Diffstat (limited to 'webui.py')
-rw-r--r--webui.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/webui.py b/webui.py
index a0f3757f..ade7334b 100644
--- a/webui.py
+++ b/webui.py
@@ -9,7 +9,7 @@ from fastapi.middleware.gzip import GZipMiddleware
from modules.paths import script_path
-from modules import devices, sd_samplers
+from modules import devices, sd_samplers, upscaler
import modules.codeformer_model as codeformer
import modules.extras
import modules.face_restoration
@@ -73,12 +73,11 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
def initialize():
- modules.scripts.load_scripts()
if cmd_opts.ui_debug_mode:
- class enmpty():
- name = None
- shared.sd_upscalers = [enmpty()]
+ shared.sd_upscalers = upscaler.UpscalerLanczos().scalers
+ modules.scripts.load_scripts()
return
+
modelloader.cleanup_models()
modules.sd_models.setup_model()
codeformer.setup_model(cmd_opts.codeformer_models_path)
@@ -86,6 +85,8 @@ def initialize():
shared.face_restorers.append(modules.face_restoration.FaceRestoration())
modelloader.load_upscalers()
+ modules.scripts.load_scripts()
+
modules.sd_models.load_model()
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model)))
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))