Put image upscaling nodes in image/upscaling category.

2 years ago · c8f1acc4eb
2 changed files with 2 additions and 2 deletions
--- a/comfy_extras/nodes_upscale_model.py
+++ b/comfy_extras/nodes_upscale_model.py
@ -35,7 +35,7 @@ class ImageUpscaleWithModel:
    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "upscale"

-    CATEGORY = "image"
+    CATEGORY = "image/upscaling"

    def upscale(self, upscale_model, image):
        device = comfy.model_management.get_torch_device()
--- a/nodes.py
+++ b/nodes.py
@ -907,7 +907,7 @@ class ImageScale:
    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "upscale"

-    CATEGORY = "image"
+    CATEGORY = "image/upscaling"

    def upscale(self, image, upscale_method, width, height, crop):
        samples = image.movedim(-1,1)