This commit is contained in:
Haoming
2026-02-05 10:15:38 +08:00
parent 6174d3c642
commit 85c832ff65
+2 -1
View File
@@ -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()