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

按下面操作即可:

操作 是否必须
更新ComfyUI 必须
更新Custom Nodes 必须
更新GGUF节点 建议
完全重启ComfyUI 必须
重新下载模型 一般不用

Reddit原帖用户反馈

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

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

结果最后:

仅仅是 ComfyUI 太旧。

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

“Z-image is FAST.”

说明模型本身没问题。


总结

这个报错本质上不是:

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

而是:

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

因此:

更新 + 重启

通常就能解决。

如果还是不行:

重点检查:

ComfyUI-GGUF

节点版本。

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐