From 4618d8006fc3620aeb2e473bb0158332adb5df5f Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Sun, 1 Dec 2024 20:06:38 -0600 Subject: [PATCH] Just in case make sure cleanup is called on motion model for SparseCtrl --- adv_control/control.py | 2 ++ 1 file changed, 2 insertions(+) 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)