From 995807b4be8197a835f8ec14cd8bb29d816fe655 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Tue, 15 Oct 2024 10:13:28 -0700 Subject: [PATCH] Improve custom node compatibility by including this stub symbol --- comfy/nodes/base_nodes.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/comfy/nodes/base_nodes.py b/comfy/nodes/base_nodes.py index 74997c895..bdb00ae28 100644 --- a/comfy/nodes/base_nodes.py +++ b/comfy/nodes/base_nodes.py @@ -21,13 +21,14 @@ from .. import sd from .. import utils from .. import clip_vision as clip_vision_module from .. import model_management -from .. import ops from ..cli_args import args from ..cmd import folder_paths, latent_preview +from ..component_model.deprecation import _deprecate_method from ..component_model.tensor_types import RGBImage, RGBImageBatch, MaskBatch from ..execution_context import current_execution_context from ..images import open_image +from ..interruption import interrupt_current_processing from ..ldm.flux.weight_dtypes import FLUX_WEIGHT_DTYPES from ..model_downloader import get_filename_list_with_downloadable, get_or_download, KNOWN_CHECKPOINTS, KNOWN_CLIP_VISION_MODELS, KNOWN_GLIGEN_MODELS, KNOWN_UNCLIP_CHECKPOINTS, KNOWN_LORAS, KNOWN_CONTROLNETS, KNOWN_DIFF_CONTROLNETS, KNOWN_VAES, KNOWN_APPROX_VAES, get_huggingface_repo_list, KNOWN_CLIP_MODELS, KNOWN_UNET_MODELS from ..nodes.common import MAX_RESOLUTION @@ -38,6 +39,11 @@ from ..sd import VAE from ..utils import comfy_tqdm +@_deprecate_method(version="0.2.3", message="Use interrupt_current_processing from comfy.interruption") +def interrupt_processing(value=True): + interrupt_current_processing(value) + + class CLIPTextEncode: @classmethod def INPUT_TYPES(s):