From 85c832ff654db75670892cbdd026199b16bee7a4 Mon Sep 17 00:00:00 2001 From: Haoming Date: Thu, 5 Feb 2026 10:15:38 +0800 Subject: [PATCH] cnet --- backend/patcher/controlnet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/patcher/controlnet.py b/backend/patcher/controlnet.py index c966de2e..ca3a01a2 100644 --- a/backend/patcher/controlnet.py +++ b/backend/patcher/controlnet.py @@ -344,7 +344,8 @@ class ControlNet(ControlBase): def cleanup(self): self.model_sampling_current = None - memory_management.unload_model(self.control_model_wrapped) + if getattr(self, "control_model_wrapped", None) is not None: + memory_management.unload_model(self.control_model_wrapped) super().cleanup()