Move some nodes out of _for_testing.

Comfy-Org · Sep 25, 2024 · 037c315 · 037c315
1 parent bdd4a22
commit 037c315
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/comfy_extras/nodes_model_downscale.py b/comfy_extras/nodes_model_downscale.py
@@ -17,7 +17,7 @@ def INPUT_TYPES(s):
     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
@@ -116,6 +116,7 @@ def INPUT_TYPES(s):
     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
@@ -154,7 +154,7 @@ def INPUT_TYPES(s):
     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
@@ -107,7 +107,7 @@ def linear_cfg(args):
         return (m, )
 
 class ImageOnlyCheckpointSave(comfy_extras.nodes_model_merging.CheckpointSave):
-    CATEGORY = "_for_testing"
+    CATEGORY = "advanced/model_merging"
 
     @classmethod
     def INPUT_TYPES(s):