ltxv: fix preprocessing exception when compression is 0.

This commit is contained in:
Michael.Kupchik 2025-03-29 14:09:45 +00:00
parent 2d17d8910c
commit 731bd6f1cb

View File

@ -446,10 +446,9 @@ class LTXVPreprocess:
CATEGORY = "image"
def preprocess(self, image, img_compression):
if img_compression > 0:
output_images = []
for i in range(image.shape[0]):
output_images.append(preprocess(image[i], img_compression))
output_images = []
for i in range(image.shape[0]):
output_images.append(preprocess(image[i], img_compression))
return (torch.stack(output_images),)