298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 05:57:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix LTXVPreprocess on resolutions that are not multiples of 2.

Browse Source
This commit is contained in:
comfyanonymous 2025-03-05 07:18:13 -05:00
parent dc134b2fdb
commit 30e6cfb1a0
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_lt.py
View File

@ -413,7 +413,7 @@ def preprocess(image: torch.Tensor, crf=29):
if crf == 0: if crf == 0:
return image return image
image_array = (image * 255.0).byte().cpu().numpy() image_array = (image[:(image.shape[0] // 2) * 2, :(image.shape[1] // 2) * 2] * 255.0).byte().cpu().numpy()
with io.BytesIO() as output_file: with io.BytesIO() as output_file:
encode_single_frame(output_file, image_array, crf) encode_single_frame(output_file, image_array, crf)
video_bytes = output_file.getvalue() video_bytes = output_file.getvalue()
@ -449,10 +449,10 @@ class LTXVPreprocess:
def preprocess(self, image, img_compression): def preprocess(self, image, img_compression):
output_image = image output_image = image
if img_compression > 0: if img_compression > 0:
output_image = torch.zeros_like(image) output_images = []
for i in range(image.shape[0]): for i in range(image.shape[0]):
output_image[i] = preprocess(image[i], img_compression) output_images.append(preprocess(image[i], img_compression))
return (output_image,) return (torch.stack(output_images),)
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {