From 669c90fec8a499b3bff7558a39db8c6c65ccaa6a Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Sun, 19 Oct 2025 13:35:27 +0800
Subject: [PATCH] k_model

---
 backend/patcher/unet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/patcher/unet.py b/backend/patcher/unet.py
index c3166754..ed8c64fd 100644
--- a/backend/patcher/unet.py
+++ b/backend/patcher/unet.py
@@ -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):