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()