mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-09 18:27:03 +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):
|
||||
samples = image.movedim(-1,1)
|
||||
#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]))
|
||||
if scale_by >= 1:
|
||||
return (image,)
|
||||
return image
|
||||
width = round(samples.shape[3] * scale_by)
|
||||
height = round(samples.shape[2] * scale_by)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user