ONE-PEACE

环境要求

• Python 3.6 至 3.10
• PyTorch >= 1.10.0（推荐 1.13.1）
• CUDA Version >= 10.2（推荐 11.6）
• 可选：Apex、xformers、FlashAttention 加速库

安装与启动步骤

1. 1克隆仓库

   从 GitHub 拉取 ONE-PEACE 源码到本地。

   git clone https://github.com/OFA-Sys/ONE-PEACE

   复制

2. 2进入项目目录

   切换到克隆下来的项目根目录，后续命令都在此目录执行。

   cd ONE-PEACE

   复制

3. 3安装依赖包

   根据 requirements.txt 安装项目所需 Python 依赖。

   pip install -r requirements.txt

   复制

4. 4可选装 Apex

   Apex 用于加速训练，非必需；需已配置好 CUDA 编译环境。

   git clone https://github.com/NVIDIA/apex
   cd apex && pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--distributed_adam" --global-option="--deprecated_fused_adam" ./

   复制

5. 5可选装 xformers

   安装 xformers 以启用内存高效注意力，非必需。

   conda install xformers -c xformers

   复制

6. 6可选装 FlashAttention

   安装 FlashAttention 以使用更快的 LayerNorm，非必需。

   git clone --recursive https://github.com/HazyResearch/flash-attention
   cd flash-attention && pip install .
   cd csrc/layer_norm && pip install .

   复制

如何确认成功

README 未给出本地启动验证命令，可用官方 HuggingFace Spaces Demo 在线验证效果。

常见问题

注意事项

• Apex、xformers、FlashAttention 均为可选，安装失败不影响基础依赖使用。
• 安装前请确认 PyTorch、CUDA 版本满足要求，避免编译类库失败。
• README 未提供本地推理/训练的具体命令，使用前请阅读 Usage 及 checkpoints.md。