RuntimeError: shape ‘[0, 6, 16]’ is invalid for input of size 72

快速结论：此错误出现在 ComfyUI 加载官方 NVIDIA PiD v1.5 检查点时。优先排查点在于，官方 Comfy-Org PiD 检查点中包含了额外的 pixel_blocks 参数（如 compress_to_attn、expand_from_attn、adaLN_modulation），而这些参数在公开的 NVIDIA 检查点中不存在。因此直接加载 NVIDIA 检查点会因状态字典结构不匹配而触发该运行时错误。

问题场景

用户在 ComfyUI 中尝试加载从官方 NVIDIA PiD v1.5 检查点转换而来的 .safetensors 文件时，在加载扩散模型阶段、推理开始前，process_unet_state_dict() 函数内部抛出该错误。

报错原文

RuntimeError: shape '[0, 6, 16]' is invalid for input of size 72

File:
comfy/supported_models.py

Function:
process_unet_state_dict()

Line:
vv = v.view(p2, 6, pixel_dim, *trail)

原因分析

根据 Issue 中的调查，该错误的直接原因是状态字典中张量形状与代码期望的形状不匹配。根本原因是：官方 Comfy-Org PiD 检查点并非简单地从 NVIDIA 检查点转换而来。Comfy-Org 模型中包含额外的 pixel_blocks 参数（例如 compress_to_attn、expand_from_attn、adaLN_modulation），这些是经过额外重新打包/导出步骤后加入的，而这一步骤当前并未公开。因此，直接加载官方 NVIDIA 检查点会导致状态字典结构不一致，从而触发形状验证失败。

环境排查

ComfyUI 版本（建议升级到最新版，Issue 关闭时间为 2026-07-13，对应版本可能包含相关支持）。
确认使用的检查点来源：是官方 Comfy-Org PiD 检查点（可正常加载），还是从 NVIDIA 官方下载的 .pth 文件转换的 .safetensors。


Python 版本（一般无特殊限制，但确保环境干净）。
显卡驱动和 PyTorch 版本（非直接原因，但建议符合 ComfyUI 通用要求）。


解决步骤

停止将官方 NVIDIA PiD v1.5 检查点直接转换为 .safetensors 加载。 当前 ComfyUI 的实现依赖于包含额外 pixel_blocks 参数的检查点结构，直接转换的 NVIDIA 检查点缺少这些参数。
使用官方 Comfy-Org PiD 检查点： 前往 Comfy-Org/PixelDiT 下载与 ComfyUI 兼容的模型文件。
（可优先尝试）等待或参与开发： 此 Issue 表明官方 NVIDIA PiD v1.5 的完整支持需要额外的转换工具或文档。建议关注 Comfy-Org/ComfyUI 仓库的后续更新或相关 PR（如关联的 #14424）。在官方提供转换路径之前，请使用 Comfy-Org 的预打包检查点。

验证方法
使用官方 Comfy-Org PiD 检查点加载模型后，如果不再抛出 RuntimeError: shape '[0, 6, 16]' is invalid for input of size 72，并且能够正常进入推理流程，即可确认问题解决。
参考来源
Comfy-Org/ComfyUI #14879


AI 工具推荐
想把多个 AI 模型放在一个入口？
GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型，适合写作、绘图、视频和日常 AI 工作流。
了解 GamsGo AI
推广链接：通过此链接购买，我可能获得佣金，不影响你的价格。

RuntimeError: shape ‘[0, 6, 16]’ is invalid for input of size 72