ComfyUI自定义节点返回多输出时IMAGE类型报错求助

核心问题定位 #

报错里的张量形状(1, 1, 1024, 3)不符合ComfyUI标准IMAGE输出格式（标准为(batch_size, height, width, 3)），多了一个冗余维度——这是PIL无法解析的直接原因，Save Image节点只认4维的(batch、高、宽、通道)张量，多余的维度会导致解析失败。

可能的触发原因 #

• 多输出修改后，VAE Decode的结果没做正确的维度转换，比如漏了把(1,3,H,W)转成(1,H,W,3)的步骤
• 返回输出的顺序和节点OUTPUT_TYPES定义不匹配，导致Save Image节点误接了其他输出张量
• 节点内部处理MODEL、CONDITIONING时，不小心污染了IMAGE变量（比如变量名冲突、误操作张量）

具体排查与修复步骤 #

1. 检查VAE Decode后的维度转换
   找到节点内VAE解码的代码段，确认是否包含标准的维度转置逻辑：

   # ComfyUI标准VAE解码后处理
   latents = ...  # 你的潜变量输出
   image = vae.decode(latents).cpu()
   # 关键：把(1,3,H,W)转成(1,H,W,3)
   image = image.permute(0, 2, 3, 1)
   image = image.clamp(0, 1)
   

   要是多输出修改时不小心删掉或注释了permute这一步，就会导致维度异常。

2. 核对返回顺序与OUTPUT_TYPES定义
   确认节点类的return语句输出顺序，和OUTPUT_TYPES里的输出顺序完全一致：

   # 示例OUTPUT_TYPES定义
   OUTPUT_TYPES = {
       "outputs": [
           ("IMAGE",),
           ("MODEL",),
           ("CONDITIONING",),
           ("VAE",),
       ]
   }
   
   # 对应的return语句必须顺序一致
   return (image, model, conditioning, vae)
   

   顺序不匹配会导致后续节点接错数据，比如把CONDITIONING当成IMAGE传入Save Image。

3. 单独验证IMAGE输出
   临时修改节点，只返回IMAGE：

   return (image,)
   

   运行后如果Save Image能正常工作，说明问题出在多输出时的其他逻辑干扰了IMAGE张量。

4. 排查其他输出的处理逻辑
   检查MODEL、CONDITIONING、VAE的代码，看是否有变量名和IMAGE重复，或者不小心对IMAGE执行了额外的维度操作（比如unsqueeze(1)）。

用户提供的报错信息 #

PIL无法处理数据类型(1, 1, 1024, 3)

内容的提问来源于stack exchange，提问作者drew15487