From e03bd9e0e38de5a420ea54f56319394d393c1649 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Sat, 15 Mar 2025 07:37:26 -0500 Subject: [PATCH] Call unpatch_hooks at the start of ModelPatcher.partially_unload --- comfy/model_patcher.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index e291158c..40f0243e 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -747,6 +747,7 @@ class ModelPatcher: def partially_unload(self, device_to, memory_to_free=0): with self.use_ejected(): + self.unpatch_hooks() memory_freed = 0 patch_counter = 0 unload_list = self._load_list()