Put image upscaling nodes in image/upscaling category.

c8f1acc4 · comfyanonymous · e33dc2b3 · c8f1acc4 · c8f1acc4
Commit c8f1acc4 authored Mar 11, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

comfy_extras/nodes_upscale_model.py comfy_extras/nodes_upscale_model.py +1 -1

nodes.py nodes.py +1 -1

No files found.
--- 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)