fix the memory leakage issue

comfy/model_patcher.py

@@ -499,12 +499,19 @@ def add_patches(self, patches, strength_patch=1.0, strength_model=1.0):
                     if len(k) > 2:
                         function = k[2]
                 org_key=key.replace("diffusion_model", "diffusion_model._orig_mod")
-                if key in model_sd or org_key in model_sd:
+                if key in model_sd:
                     p.add(k)
                     current_patches = self.patches.get(key, [])
                     current_patches.append((strength_patch, patches[k], strength_model, offset, function))
                     self.patches[key] = current_patches
                     self.patches[org_key] = current_patches
+                elif org_key in model_sd:
+                    if key in self.patches:
+                        self.patches.pop(key)
+                    p.add(k)
+                    current_patches = self.patches.get(org_key, [])
+                    current_patches.append((strength_patch, patches[k], strength_model, offset, function))
+                    self.patches[org_key] = current_patches
 
             self.patches_uuid = uuid.uuid4()
             return list(p)

comfy_extras/nodes_torch_compile.py

@@ -19,7 +19,7 @@ def INPUT_TYPES(s):
                 "backend": (["inductor", "cudagraphs", "openvino"],),
             },
             "optional": {
-                "openvino_device": (available_devices,),
+                "openvino device": (available_devices,),
             },
         }
 
@@ -30,6 +30,7 @@ def INPUT_TYPES(s):
     EXPERIMENTAL = True
 
     def patch(self, model, backend, openvino_device):
+        print(model.__class__.__name__)
         if backend == "openvino":
             options = {"device": openvino_device}
             try:
@@ -39,6 +40,12 @@ def patch(self, model, backend, openvino_device):
                     "Could not import openvino python package. "
                     "Please install it with `pip install openvino`."
                 )
+            import openvino.frontend.pytorch.torchdynamo.execute as ov_ex
+
+            torch._dynamo.reset()
+            ov_ex.compiled_cache.clear()
+            ov_ex.req_cache.clear()
+            ov_ex.partitioned_modules.clear()
         else:
             options = None
         m = model.clone()
@@ -55,4 +62,4 @@ def patch(self, model, backend, openvino_device):
 
 NODE_CLASS_MAPPINGS = {
     "TorchCompileModel": TorchCompileModel,
-}
+}