From 395e2fd23e2f7ea994941abf3b551a693358d991 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 9 Jan 2024 15:20:19 -0500 Subject: [PATCH] further hijack cleanup --- extensions-builtin/Lora/networks.py | 4 +--- webui.py | 5 ++--- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 4a4198e1a..b0a34b06d 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -16,7 +16,7 @@ import network_glora import lora_convert import torch import diffusers.models.lora -from modules import shared, devices, sd_models, sd_models_compile, errors, scripts, sd_hijack +from modules import shared, devices, sd_models, sd_models_compile, errors, scripts debug = os.environ.get('SD_LORA_DEBUG', None) is not None @@ -209,8 +209,6 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No net.unet_multiplier = unet_multipliers[i] if unet_multipliers else 1.0 net.dyn_dim = dyn_dims[i] if dyn_dims else 1.0 loaded_networks.append(net) - if failed_to_load_networks: - sd_hijack.model_hijack.comments.append("Networks not found: " + ", ".join(failed_to_load_networks)) while len(lora_cache) > shared.opts.lora_in_memory_limit: name = next(iter(lora_cache)) diff --git a/webui.py b/webui.py index 60a68a7dc..316021776 100644 --- a/webui.py +++ b/webui.py @@ -58,9 +58,8 @@ fastapi_args = { } } -if shared.backend == shared.Backend.ORIGINAL: - import modules.sd_hijack - timer.startup.record("ldm") +import modules.sd_hijack +timer.startup.record("ldm") modules.loader.initialized = True