ImportError: No module named ‘triton’

快速结论：该报错并非 Python 模块缺失，而是在 ComfyUI 中使用 CLIPLoader 节点加载 Qwen 文本编码器时，因 safetensors 文件大小与元数据中声明的 shape 不匹配而触发的 RuntimeError。优先排查模型文件是否损坏，以及是否使用了正确的节点类型（Diffusion Loader 而非 UNet Loader）。

问题场景

用户在 ComfyUI 0.27.0 中通过 CLIPLoader 节点加载 qwen_3_4b.safetensors 文件时触发报错。该文件位于 Hugging Face 的 Comfy-Org/z_image 仓库的 split_files/text_encoders/ 目录下。

报错原文

RuntimeError: shape '[151936, 2560]' is invalid for input of size 301956086

完整堆栈中的核心错误行：

tensor = torch.frombuffer(mv[start:end], dtype=_TYPES[info["dtype"]]).view(info["shape"])

原因分析

该错误表明 safetensors 文件内部的某个 tensor 元数据中声明的形状（shape）[151936, 2560] 所对应的总元素数应为 151936 × 2560 = 388,956,160，但实际文件数据段的大小仅为 301,956,086 字节。导致 torch.frombuffer 在 reshape 时发现元素数量不匹配。

可能原因包括：

模型文件下载不完整或损坏（开发者首先建议重新下载）。
文件来源错误：用户可能使用了错误的文件（如 UNet 模型而非文本编码器）。
节点选择错误：应使用“Diffusion Loader”节点加载此类模型，而非“UNet Loader”或“CLIPLoader”。

环境排查

ComfyUI 版本：0.27.0
Python 版本：3.13.12
PyTorch 版本：2.10.0+cu130
CUDA 设备：NVIDIA GeForce RTX 5060 Ti（VRAM 8 GB）
操作系统：Windows 10/11（win32）
模型文件：qwen_3_4b.safetensors（大小约 ~300 MB）
节点类型：CLIPLoader（节点 ID: 18）

解决步骤

步骤 1：下载官方推荐的模型文件
从 Hugging Face 指定地址重新下载 qwen_3_4b.safetensors 文件：
https://huggingface.co/Comfy-Org/z_image/blob/main/split_files/text_encoders/qwen_3_4b.safetensors
确保下载完整，可对比文件大小（约 301 MB 左右）。
步骤 2：检查下载文件完整性
如果再次下载后问题依旧，使用工具（如 sha256sum 或 Windows 自带的 certutil）校验文件哈希值，确认与官方源一致。
步骤 3：更换正确节点类型
在 ComfyUI 工作流中，尝试使用 Diffusion Loader 节点（而非 CLIPLoader 或 UNet Loader）来加载该模型。
操作路径：右键添加节点 → Advanced / Loaders / Diffusion Loader。
步骤 4：清理 ComfyUI 缓存
删除 ComfyUI/cache/ 目录（如果有），重启 ComfyUI 后重新加载模型。

验证方法

加载模型后，检查 CLIPLoader 或 Diffusion Loader 节点是否正常输出条件嵌入（conditioning），且控制台无新错误。可尝试简单的 txt2img 生成操作，观察是否正常完成。

参考来源

Comfy-Org/ComfyUI #14808

AI 工具推荐

想把多个 AI 模型放在一个入口？

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型，适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接：通过此链接购买，我可能获得佣金，不影响你的价格。

ImportError: No module named ‘triton’