CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

k_model

Browse Source
This commit is contained in:
Haoming 2025-10-19 13:35:27 +08:00
parent 6b2483ce29
commit 669c90fec8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/patcher/unet.py
View File

@ -207,7 +207,7 @@ class NunchakuPatcher(UnetPatcher):
return self.model
def forge_unpatch_model(self, target_device=None):
if target_device is not None:
if target_device is not None and hasattr(self.model, "diffusion_model"): # k_model/KModel/cleanup()
self.model.diffusion_model.to(target_device)
def to_load_list(self):