From 037c3159b62717f4d50e50afff1d0275475e2499 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Wed, 25 Sep 2024 03:24:13 -0400 Subject: [PATCH] Move some nodes out of _for_testing. --- comfy_extras/nodes_model_downscale.py | 2 +- comfy_extras/nodes_stable_cascade.py | 1 + comfy_extras/nodes_tomesd.py | 2 +- comfy_extras/nodes_video_model.py | 2 +- 4 files changed, 4 insertions(+), 3 deletions(-) diff --git a/comfy_extras/nodes_model_downscale.py b/comfy_extras/nodes_model_downscale.py index 58b5073e..15ffc4c8 100644 --- a/comfy_extras/nodes_model_downscale.py +++ b/comfy_extras/nodes_model_downscale.py @@ -17,7 +17,7 @@ class PatchModelAddDownscale: RETURN_TYPES = ("MODEL",) FUNCTION = "patch" - CATEGORY = "_for_testing" + CATEGORY = "model_patches/unet" def patch(self, model, block_number, downscale_factor, start_percent, end_percent, downscale_after_skip, downscale_method, upscale_method): model_sampling = model.get_model_object("model_sampling") diff --git a/comfy_extras/nodes_stable_cascade.py b/comfy_extras/nodes_stable_cascade.py index fcbbeb27..00340321 100644 --- a/comfy_extras/nodes_stable_cascade.py +++ b/comfy_extras/nodes_stable_cascade.py @@ -116,6 +116,7 @@ class StableCascade_SuperResolutionControlnet: RETURN_NAMES = ("controlnet_input", "stage_c", "stage_b") FUNCTION = "generate" + EXPERIMENTAL = True CATEGORY = "_for_testing/stable_cascade" def generate(self, image, vae): diff --git a/comfy_extras/nodes_tomesd.py b/comfy_extras/nodes_tomesd.py index df048506..ce7b32c7 100644 --- a/comfy_extras/nodes_tomesd.py +++ b/comfy_extras/nodes_tomesd.py @@ -154,7 +154,7 @@ class TomePatchModel: RETURN_TYPES = ("MODEL",) FUNCTION = "patch" - CATEGORY = "_for_testing" + CATEGORY = "model_patches/unet" def patch(self, model, ratio): self.u = None diff --git a/comfy_extras/nodes_video_model.py b/comfy_extras/nodes_video_model.py index 7f2146d1..e7a7ec18 100644 --- a/comfy_extras/nodes_video_model.py +++ b/comfy_extras/nodes_video_model.py @@ -107,7 +107,7 @@ class VideoTriangleCFGGuidance: return (m, ) class ImageOnlyCheckpointSave(comfy_extras.nodes_model_merging.CheckpointSave): - CATEGORY = "_for_testing" + CATEGORY = "advanced/model_merging" @classmethod def INPUT_TYPES(s):