ltxv: fix preprocessing exception when compression is 0. (#7431)

This commit is contained in:
Michael Kupchick 2025-03-30 03:03:02 +03:00 committed by GitHub
parent 2d17d8910c
commit 832fc02330
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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),)