nodes/set-soft-ComfyUI-ImageMisc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Fixed] upscale/upscale_comfy

Browse Source
This commit is contained in:
Salvador E. Tropea
2025-11-19 10:23:31 -03:00
parent 6bc0a90de4
commit 8b1b6431ef
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/nodes/helpers.py
+6 -6
View File
@@ -29,20 +29,20 @@ def empty_image(b=0, h=64, w=64, c=1):
return torch.zeros(dims, dtype=torch.float32, device="cpu")
def upscale(image, width, height, upscale_method):
def upscale(image, width, height, upscale_method, crop="disabled"):
# return F.interpolate(image, size=(height, width), mode=upscale_method)
return common_upscale(image, width, height, upscale_method, crop="disabled")
return common_upscale(image, width, height, upscale_method, crop=crop)
def upscale_comfy(image, width, height, upscale_method, crop="disabled"):
if image.dim == 3:
if image.dim() == 3:
# A mask
if upscale_method == "lanczos":
# Lanczos needs an RGB image
return upscale(image.unsqueeze(1).repeat(1, 3, 1, 1), width, height, upscale_method,
crop=crop).movedim(1, -1)[:, :, :, 0]
return common_upscale(image.unsqueeze(1).repeat(1, 3, 1, 1), width, height, upscale_method,
crop=crop).movedim(1, -1)[:, :, :, 0]
else:
return upscale(image.unsqueeze(1), width, height, upscale_method, crop=crop).squeeze(1)
return common_upscale(image.unsqueeze(1), width, height, upscale_method, crop=crop).squeeze(1)
return common_upscale(image.movedim(-1, 1), width, height, upscale_method, crop=crop).movedim(1, -1)