mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-23 04:40:15 +08:00
Merge branch 'comfyanonymous:master' into master
This commit is contained in:
commit
b7d9be6864
@ -1000,7 +1000,6 @@ class WAN21(BaseModel):
|
|||||||
device = kwargs["device"]
|
device = kwargs["device"]
|
||||||
|
|
||||||
if image is None:
|
if image is None:
|
||||||
image = torch.zeros_like(noise)
|
|
||||||
shape_image = list(noise.shape)
|
shape_image = list(noise.shape)
|
||||||
shape_image[1] = extra_channels
|
shape_image[1] = extra_channels
|
||||||
image = torch.zeros(shape_image, dtype=noise.dtype, layout=noise.layout, device=noise.device)
|
image = torch.zeros(shape_image, dtype=noise.dtype, layout=noise.layout, device=noise.device)
|
||||||
|
|||||||
@ -446,10 +446,9 @@ class LTXVPreprocess:
|
|||||||
CATEGORY = "image"
|
CATEGORY = "image"
|
||||||
|
|
||||||
def preprocess(self, image, img_compression):
|
def preprocess(self, image, img_compression):
|
||||||
if img_compression > 0:
|
output_images = []
|
||||||
output_images = []
|
for i in range(image.shape[0]):
|
||||||
for i in range(image.shape[0]):
|
output_images.append(preprocess(image[i], img_compression))
|
||||||
output_images.append(preprocess(image[i], img_compression))
|
|
||||||
return (torch.stack(output_images),)
|
return (torch.stack(output_images),)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user