lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-01 11:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 17:31:11 -08:00
parent eff0e978a8
commit cc49e13356
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/forge_legacy_preprocessors/annotator/pidinet/__init__.py
View File

@@ -6,7 +6,7 @@ from annotator.pidinet.model import pidinet
from annotator.util import safe_step
from modules import devices
from annotator.annotator_path import models_path
from scripts.utils import load_state_dict
from ldm_patched.modules.utils import load_torch_file
netNetwork = None
remote_model_path = "https://huggingface.co/lllyasviel/Annotators/resolve/main/table5_pidinet.pth"
@@ -24,9 +24,9 @@ def apply_pidinet(input_image, is_safe=False, apply_fliter=False):
from modules.modelloader import load_file_from_url
load_file_from_url(remote_model_path, model_dir=modeldir)
netNetwork = pidinet()
ckp = load_state_dict(modelpath)
ckp = load_torch_file(modelpath)
netNetwork.load_state_dict({k.replace('module.',''):v for k, v in ckp.items()})
netNetwork = netNetwork.to(devices.get_device_for("controlnet"))
netNetwork.eval()
assert input_image.ndim == 3