From 2ecb869f31f4abab5922c1bd611e375d5bb28e8e Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 23 Feb 2024 20:43:06 -0800 Subject: [PATCH] try solve #386 --- .../sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py b/extensions-builtin/sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py index f377ccdd..15d8f809 100644 --- a/extensions-builtin/sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py +++ b/extensions-builtin/sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py @@ -7,12 +7,12 @@ from modules_forge.supported_controlnet import ControlModelPatcher from modules_forge.forge_sampler import sampling_prepare from ldm_patched.modules.utils import load_torch_file from ldm_patched.modules import model_patcher -from ldm_patched.modules.model_management import cast_to_device, LoadedModel, current_loaded_models +from ldm_patched.modules.model_management import cast_to_device, current_loaded_models from ldm_patched.modules.lora import model_lora_keys_unet def is_model_loaded(model): - return LoadedModel(model) in current_loaded_models + return any(model == m.model for m in current_loaded_models) class InpaintHead(torch.nn.Module):