onediff

环境要求

• Python 3 环境（README 全程使用 python3 -m pip）
• 已安装 pip，可正常访问 PyPI
• NVIDIA GPU + CUDA 环境（README 按 CUDA 11.8 / 12.1 / 12.2 区分安装包）
• torch、transformers、diffusers 运行依赖
• 可访问 GitHub 或 README 给出的国内镜像源以下载 OneFlow

安装与启动步骤

1. 1安装基础依赖

   先装 torch、transformers 与 diffusers，README 提示 diffusers/transformers 可自行选择想要的版本。

   python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3"

   复制

2. 2安装 Nexfort 后端

   Nexfort 与 OneFlow 二选一；DiT 结构模型或 H100 设备推荐用 Nexfort，无需两者都装。

   python3 -m pip install -U torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 torchao==0.1
   python3 -m pip install -U nexfort

   复制

3. 3安装 OneFlow 后端

   备选后端，适用于非 DiT 模型和非 H100 场景。此处以 CUDA 11.8 为例，其它 CUDA 版本请按 README 更换链接。

   python3 -m pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/community_cu118

   复制

4. 4从 PyPI 装 onediff

   推荐方式，直接安装 onediff 预发布包，无需克隆仓库。

   python3 -m pip install --pre onediff

   复制

5. 5从源码安装

   需要改代码或跟最新进展时使用；开发模式用 pip install -e '.[dev]'。

   git clone https://github.com/siliconflow/onediff.git
   cd onediff && python3 -m pip install -e .

   复制

如何确认成功

README 未给出启动或自检命令，以各步 pip 安装无报错、且 onediff 安装完成作为成功标志。

常见问题

注意事项

• README 更新 OneFlow 较频繁，务必使用它给出的链接安装，不要随意用默认源版本
• CUDA 11.8 与 12.1/12.2 使用不同的安装链接，选错会导致后端不可用
• 需要 NVIDIA GPU 环境，OneDiff 主要针对 NVIDIA GPU 做性能调优
• 本教程只覆盖安装部分，Python API 与 ComfyUI/Diffusers 用法请参考官方 Wiki