摩尔线程S80适配ComfyUI运行Z-Image-Turbo文本生成图片模型

[流水U]

当前流行的可视化AI创作工作流ComfyUI一直没有可用的适配摩尔线程S80的版本。最近开源的几款文本生成图片的模型在社区非常火热，于是趁着热度，也想在S80上体验一下，就尝试基于ComfyUI release v0.3.75版本做了一些修改适配工作，调试后成功运行。

下载ComfyUI-musa

我的ComfyUI-musa开源项目地址：https://gitee.com/uniqq/ComfyUI-musa
可使用git clone

cd ~
git clone https://gitee.com/uniqq/ComfyUI-musa.git

安装配置可参考项目仓库首页文档。

安装S80显卡驱动和SDK

我使用的是OS版本是Ubuntu 22.04.5 LTS，内核版本为5.15.0-105-generic，这也是S80驱动安装文档说明中支持的版本。
MUSA_SDK包版本为：4.3.0.CC2.1
可在摩尔线程官网下载页面进行下载，地址为：https://developer.mthreads.com/sdk/download/musa?equipment=MTT%20S80&os=Ubuntu&driverVersion=&version=4.3.0
参考SDK包内的安装文档，依次安装好显卡驱动MT_Linux_Driver、MUSA_Toolkit和muDNN库和MCCL库。

配置运行环境

安装miniconda并创建一个Python 3.10的环境：

conda create --name comfyui python=3.10

激活新创建的conda环境，并安装项目依赖的包

conda activate comfyui
cd ~/ComfyUI-musa
pip install -r requirements.txt

安装PyTorch on MUSA

官方开源项目仓库地址：https://github.com/MooreThreads/torch_musa
可以直接下载编译好的release包，建议使用torch_musa Release v2.1.0/v2.1.1版本：https://github.com/MooreThreads/torch_musa/releases/download/v2.1.0/S80-wheels.tar.gz
解压后使用pip安装

pip install torch-2.5.0-cp310-cp310-linux_x86_64.whl
pip install torch_musa-2.1.0-cp310-cp310-linux_x86_64.whl

下载Z-Image-Turbo模型文件

下载Comfy-Org制作好的适用于ComfyUI的模型文件，下载页面地址：https://modelscope.cn/models/Comfy-Org/z_image_turbo/files
下载后按以下目录结构放置好模型文件：

📂 ComfyUI-musa/
├── 📂 models/
│   ├── 📂 text_encoders/
│   │      └── qwen_3_4b.safetensors
│   ├── 📂 diffusion_models/
│   │      └── z_image_turbo_bf16.safetensors
│   └── 📂 vae/
│          └── ae.safetensors

运行ComfyUI-musa

cd ~/ComfyUI-musa
python main.py --listen --port 8080

通过浏览器访问http://[实际ip]:8080即可打开ComfyUI的页面，选择左侧“模板”菜单，在弹出的模板中选择"Z-Image-Turbo 文生图"工作流模板，输入提示词，其它参数均可默认，点击页面右上角“运行”按钮即可体验。
按默认的参数，首次运行生成1张512*768的图片，9次迭代耗时约80多秒：

不得不说，这次Tongyi-MAI开源的Z-Image-Turbo使用效果真是超预期，非常棒
:) have fun~

[weixin_44117212]

我尝试按照你的流程安装，最终成功了，总结一下子：

坑1、下载的torch_musa Release v2.1.0/v2.1.1里面只有torch和torch-musa的两个包，缺少了torchvision和torchaudio的musa专项包。以至于安装torch时就提示出错。在GitHub的musa站点上也未找到匹配的whl下载。可能需要编译吧，但实在太麻烦了。后来我直接下载2.5.0版，高低都出错，那就用个新的。

坑2、MUSA_SDK 4.3.0.CC2.1 里面是没有 mcll 的，最终找了4.3.0.CC2.2版里给 S4000 的包凑数，CloudNative_Toolkits_v2.0.0也是另找链接下载的，至此 confyui 才最终跑起来。然而我两张S80显卡，实测 mudnn 和 mcll 其实根本没用处，满屏 error 死给你看。

坑3、折腾半天，Z-image倒是可以运行，但很多其它流程都会碰到莫名错误，要么nampy版本不对，要么C++出错，装个插件，一个不当心，就可能把torch-musa给替换成cuda的了，很头大啊。

坑4、你的方案是全手动，所以我还试了用musa-deploy来简化安装所有环境，严格按照软硬件配置来的是没问题的。最大的坑在于，这玩意甜蜜的严格遵守musa+intel的死规则，换台AMD3900的机器，他竟然报cpu错直接不给装，只能手动安装各项，那帮开发者真是天马的死硬坑货。

坑5、comfyui版本低，我去GitHub网站发现了个小玩意，torchada，装了这个竟然可以直接下载最新的官方comfyui来用，它的原理好像是劫持所有comfyui的指令，转译给musa来算，但大前提是，安装的 torch 必须是纯cpu版的，所以版本混淆到头疼，只能装低版本torch。用这个Z-image也可以跑，但毛病也和你这个修改版一样，改换别的流程或功能就是各种出错……对了，它这个要改一下main.py，整体改为32位的，因为musa显卡不支持64位的采样？

总之，musa 的 torch 开发还是有得等。
再过一年半载的，S80这波就完全过去了吧？
没估计也就没有后续玩家了。
希望他们大生意能做得好吧。
没热情了，不指望了。

[weixin_44117212]

啊啊！对了！
突然想到，前不久安装了openclaw，我要不试试全部用Ai来调制一番呢？
这个主意咋样？
……
对啊，让Ai头疼去，我又天马的有动力了！