mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
cnet
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user