wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-06 09:47:35 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

fix torch.cat requiring inputs to all be same dimensions (#12673)

Browse Source
This commit is contained in:
pythongosssss 2026-02-27 16:13:24 +00:00 committed by GitHub
parent 35e9fce775
commit 0a7f8e11b6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
comfy_extras/nodes_glsl.py
View File

@ -865,14 +865,15 @@ class GLSLShader(io.ComfyNode):
cls, image_list: list[torch.Tensor], output_batch: torch.Tensor cls, image_list: list[torch.Tensor], output_batch: torch.Tensor
) -> dict[str, list]: ) -> dict[str, list]:
"""Build UI output with input and output images for client-side shader execution.""" """Build UI output with input and output images for client-side shader execution."""
combined_inputs = torch.cat(image_list, dim=0) input_images_ui = []
input_images_ui = ui.ImageSaveHelper.save_images( for img in image_list:
combined_inputs, input_images_ui.extend(ui.ImageSaveHelper.save_images(
filename_prefix="GLSLShader_input", img,
folder_type=io.FolderType.temp, filename_prefix="GLSLShader_input",
cls=None, folder_type=io.FolderType.temp,
compress_level=1, cls=None,
) compress_level=1,
))
output_images_ui = ui.ImageSaveHelper.save_images( output_images_ui = ui.ImageSaveHelper.save_images(
output_batch, output_batch,