jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Put image upscaling nodes in image/upscaling category.

Browse Source
This commit is contained in:
comfyanonymous
2023-03-11 18:10:36 -05:00
parent e33dc2b33b
commit c8f1acc4eb
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_upscale_model.py
View File

@@ -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()

2
nodes.py
View File

@@ -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)