soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

try solve #386

Browse Source
This commit is contained in:
lllyasviel 2024-02-23 20:43:06 -08:00
parent a844834193
commit 2ecb869f31
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/sd_forge_fooocus_inpaint/scripts/forge_fooocus_inpaint.py
View File

@ -7,12 +7,12 @@ from modules_forge.supported_controlnet import ControlModelPatcher
from modules_forge.forge_sampler import sampling_prepare from modules_forge.forge_sampler import sampling_prepare
from ldm_patched.modules.utils import load_torch_file from ldm_patched.modules.utils import load_torch_file
from ldm_patched.modules import model_patcher 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 from ldm_patched.modules.lora import model_lora_keys_unet
def is_model_loaded(model): 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): class InpaintHead(torch.nn.Module):