如果您在 Mac 上使用 ComfyUI 创建精美的 AI 图像,那么在尝试将它们放大时,您可能会遇到瓶颈。具体来说,如果您尝试生成大于 1920×1920 像素的图像,您可能会看到进程在最后崩溃,并出现一条奇怪的错误消息。
这是一个令人沮丧的问题,但别担心,有一个简单的解决办法!
问题:“断言失败”错误
当 ComfyUI 几乎完成生成大图像时,它会突然停止,并且您会在终端中看到类似这样的错误消息:
简单来说,这是 PyTorch 最新版本的一个 bug。ComfyUI 正是利用 PyTorch 库在 Apple Silicon(M1、M2、M3 芯片)上实现 AI 魔法。PyTorch 中处理 Mac 图形芯片上图像数据的部分(Metal Performance Shaders,简称 MPS)无法处理超大尺寸图像的尺寸。 /AppleInternal/Library/BuildRoots/ .../MPSNDArray.mm: 829 : failed assertion \`... Error: NDArray dimension length \> INT \_MAX'
解决方案:降级你的 PyTorch 版本
好消息是,你可以通过回到稍旧、更稳定的 PyTorch 版本来解决这个问题。关键是安装2.4.0版本。
具体操作如下:
这个命令告诉 Python 的包管理器, pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0
pip
,卸载当前的 PyTorch 并安装已知没有此错误的特定版本。
安装完成后,重新启动 ComfyUI,您现在应该能够生成那些巨大的图像而不会出现任何崩溃。🎉
经过测试并确认
我已在配备 64GB RAM 的 M1 Max 上亲自测试了此修复程序,成功生成了 3000×3000 像素的图像,且未出现任何崩溃。在 PyTorch 降级之前,ComfyUI 在尝试生成此分辨率的图像时会持续崩溃。安装 PyTorch 2.4.0 后,相同的工作流程已成功完成。
需要不同的版本吗?
此错误与 PyTorch 的特定版本相关。如果您出于任何原因需要查找其他早期版本,PyTorch 官方网站上有一个有用的存档。
- 在此处查看旧版本的 PyTorch: https://pytorch.org/get-started/previous-versions/
对于那些对技术细节感兴趣的人,可以关注 GitHub 上的错误报告:
快乐生成!
原文: https://atlassc.net/2025/07/09/comfyui-crashing-on-mac-with-big-images