nodes/Kosinkadink-ComfyUI-Advanced-ControlNet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Just in case make sure cleanup is called on motion model for SparseCtrl

Browse Source
This commit is contained in:
Jedrzej Kosinski
2024-12-01 20:06:38 -06:00
parent 37b318debc
commit 4618d8006f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
adv_control/control.py
+2
View File
@@ -481,6 +481,8 @@ class SparseCtrlAdvanced(ControlNetAdvanced):
self.model_latent_format = None self.model_latent_format = None
self.local_sparse_idxs = None self.local_sparse_idxs = None
self.local_sparse_idxs_inverse = None self.local_sparse_idxs_inverse = None
if self.motion_model is not None:
self.motion_model.cleanup()
def copy(self): def copy(self):
c = SparseCtrlAdvanced(self.control_model, self.motion_model, self.timestep_keyframes, self.sparse_settings, self.global_average_pooling, self.load_device, self.manual_cast_dtype) c = SparseCtrlAdvanced(self.control_model, self.motion_model, self.timestep_keyframes, self.sparse_settings, self.global_average_pooling, self.load_device, self.manual_cast_dtype)