comfyui Z-Image运行报错：error in loading state dict for llama2 size mismatch 的解决方法

comfyui Z-Image运行报错：error in loading state dict for llama2 size mismatch 的解决方法

SEO关键词：Z-Image报错、ComfyUI报错、llama2 size mismatch、Z-Image无法运行、ComfyUI GGUF错误、Stable Diffusion教程、AI绘图报错修复

在这里插入图片描述

最近我在本地测试 Z-Image 工作流时，刚下载完模型直接点击运行，结果立刻弹出一大串报错：

error in loading state dict for llama2:
size mismatch for model.layer

一开始我还以为是模型下载损坏、工作流不兼容，甚至重新下载了模型和节点，折腾了半天。

后来发现问题其实非常简单：

问题原因

核心原因是：

ComfyUI 版本过旧。

Z-Image 使用了新的模型结构，而旧版 ComfyUI 或旧版 GGUF 节点无法正确加载新模型参数，所以会出现：

size mismatch

这种参数维度不匹配错误。

这类错误本质上是：

• 新模型结构
• 老代码加载
• 参数shape不同
• 导致模型初始化失败

类似于：

你用 Vue3 的代码去跑 Vue2 插件。

接口名字一样，但内部结构已经变了。

解决方法

方法1：更新 ComfyUI（最有效）

这是 Reddit 原帖中确认有效的方法。

进入：

ComfyUI Manager

然后执行：

Update ComfyUI

更新完成后：

一定要彻底重启 ComfyUI

很多人忽略了这一点。

仅更新不重启：

• 新代码不会加载
• Python缓存还在
• 节点还是旧版本

效果等于没更新。

正确更新流程

建议按下面顺序：

1. 更新ComfyUI

Update ComfyUI

2. 更新所有Custom Nodes

Update All Custom Nodes

尤其是：

ComfyUI-GGUF

这个节点很关键。

3. 关闭ComfyUI

不要只是刷新网页。

要：

• 关闭终端
• 关闭bat窗口
• 停止python进程

4. 重新启动

重新运行：

run_nvidia_gpu.bat

或者：

python main.py

如果更新后仍然报错

有些用户更新后还会看到：

UnetLoaderGGUF

Error(s) in loading state_dict for NextDiT:

size mismatch for x_pad_token

例如：

shape torch.Size([3840])
vs
torch.Size([1, 3840])

这属于：

GGUF节点兼容问题

并不是模型坏了。

目前主要和：

ComfyUI-GGUF

节点版本有关。

Reddit讨论里也提到了：

相关Issue：

https://github.com/city96/ComfyUI-GGUF/issues/379

目前部分版本还没有完全修复。

解决GGUF兼容问题的方法

方法1：重新拉取GGUF节点

删除：

ComfyUI/custom_nodes/ComfyUI-GGUF

然后重新安装最新版。

方法2：Git更新

进入目录：

custom_nodes/ComfyUI-GGUF

执行：

git pull

方法3：使用官方推荐工作流

有些第三方工作流：

• 节点版本老
• 模型版本不一致
• 参数结构已变化

容易导致：

size mismatch

尽量使用：

• 官方workflow
• 最新workflow
• 对应模型版本

为什么会出现 size mismatch？

这是 PyTorch 常见错误。

例如：

旧模型：

torch.Size([3840])

新版模型：

torch.Size([1, 3840])

虽然看起来差不多：

但：

• 一个是一维
• 一个是二维

模型层结构已经变化。

所以旧版加载器无法读取。

最终解决方案总结

如果你运行 Z-Image 出现：

error in loading state dict for llama2

按下面操作即可：

Reddit原帖用户反馈

很多人被这个问题卡了很久：

“我花了2小时重新下载模型和工作流。”

结果最后：

仅仅是 ComfyUI 太旧。

还有用户更新后成功运行：

“Z-image is FAST.”

说明模型本身没问题。

总结

这个报错本质上不是：

• 显卡问题
• CUDA问题
• 模型损坏

而是：

模型版本和ComfyUI代码版本不匹配。

因此：

更新 + 重启

通常就能解决。

如果还是不行：

重点检查：

ComfyUI-GGUF

节点版本。