DiffSynth-Studio MUSA Patch 入门

项目简介

DiffSynth-Studio MUSA Patch 是 DiffSynth-Studio 在摩尔线程 MUSA GPU 上的适配补丁。它在保持 DiffSynth-Studio 原有调用方式的同时，将运行时设备 和算子调用适配到 MUSA 后端，便于在 MUSA 环境中进行扩散模型推理与训练。

本文档包含补丁的基本说明、环境要求和使用方法。首次使用时，建议先阅读 DiffSynth-Studio MUSA Patch 使用文档。

快速导航

• 使用文档：查看功能特性、安装、推理和训练示例。
• 项目首页：了解上游 DiffSynth-Studio 项目。

环境准备

使用前请准备：

• 已安装并正常工作的 MUSA 驱动、MUSA SDK，以及与补丁匹配的 torch_musa；
• 可访问 DiffSynth-Studio 及其模型、数据集依赖的 Python 环境；
• MUSA GPU 设备及足够的显存。

可以先在目标环境中确认 MUSA 是否可用：

import torch
import torch_musa

print(torch.musa.is_available())

如果输出为 True，即可继续按照使用文档安装补丁并运行示例。