fix: 🚀 use the comfy util to handle graph interruption

nodes/faceswap.py

@@ -16,6 +16,8 @@
 from ..utils import pil2tensor, tensor2pil
 from ..log import mklog, NullWriter
 import sys
+import comfy.model_management as model_management
+
 
 # endregion
 
@@ -100,6 +102,7 @@ def swap(
         debug="false",
     ):
         def do_swap(img):
+            model_management.throw_exception_if_processing_interrupted()
             img = tensor2pil(img)
             ref = tensor2pil(reference)
             face_ids = {

nodes/image_interpolation.py

@@ -13,6 +13,7 @@
 import comfy
 
 import tensorflow as tf
+import comfy.model_management as model_management
 
 
 class LoadFilmModel:
@@ -104,6 +105,7 @@ def do_interpolation(
             out_tensors.append(
                 torch.from_numpy(frame) if isinstance(frame, np.ndarray) else frame
             )
+            model_management.throw_exception_if_processing_interrupted()
             pbar.update(1)
 
         out_tensors = torch.cat([tens.unsqueeze(0) for tens in out_tensors], dim=0)
@@ -228,6 +230,7 @@ def export_prores(
         process = subprocess.Popen(command, stdin=subprocess.PIPE)
 
         for frame in frames:
+            model_management.throw_exception_if_processing_interrupted()
             process.stdin.write(frame.tobytes())
 
         process.stdin.close()

nodes/image_processing.py

@@ -14,6 +14,8 @@
 from PIL.PngImagePlugin import PngInfo
 import json
 import os
+import comfy.model_management as model_management
+
 
 try:
     from cv2.ximgproc import guidedFilter
@@ -514,6 +516,7 @@ def resize_image(
         resample="lanczos",
         mask=None,
     ) -> torch.Tensor:
+        model_management.throw_exception_if_processing_interrupted()
         batch_count = 1
         img = tensor2pil(image)