mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-10 05:07:06 +08:00
fix wrong output type
This commit is contained in:
parent
8d414a99de
commit
68838ea6f2
@ -22,10 +22,10 @@ import math
|
|||||||
def downscale_input(image):
|
def downscale_input(image):
|
||||||
samples = image.movedim(-1,1)
|
samples = image.movedim(-1,1)
|
||||||
#downscaling input images to roughly the same size as the outputs
|
#downscaling input images to roughly the same size as the outputs
|
||||||
total = int(1024 * 1024)
|
total = int(1536 * 1024)
|
||||||
scale_by = math.sqrt(total / (samples.shape[3] * samples.shape[2]))
|
scale_by = math.sqrt(total / (samples.shape[3] * samples.shape[2]))
|
||||||
if scale_by >= 1:
|
if scale_by >= 1:
|
||||||
return (image,)
|
return image
|
||||||
width = round(samples.shape[3] * scale_by)
|
width = round(samples.shape[3] * scale_by)
|
||||||
height = round(samples.shape[2] * scale_by)
|
height = round(samples.shape[2] * scale_by)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user