diff --git a/adv_control/control.py b/adv_control/control.py index a6714e8..089509e 100644 --- a/adv_control/control.py +++ b/adv_control/control.py @@ -481,6 +481,8 @@ class SparseCtrlAdvanced(ControlNetAdvanced): self.model_latent_format = None self.local_sparse_idxs = None self.local_sparse_idxs_inverse = None + if self.motion_model is not None: + self.motion_model.cleanup() 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)