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

流水U 2025-12-05 21:58:17

当前流行的可视化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多秒：