TRL documentation
安装
入门指南
概念指南
Trainers
操作指南
命令行界面(CLI)使用 Jobs 进行训练自定义训练减少内存使用加速训练分布式训练在 AWS Trainium 上进行训练在 AMD GPU 上进行训练训练超过 100 万个 Token使用训练好的模型
集成
示例
API
实验性
You are viewing main version, which requires installation from source. If you'd like
regular pip install, checkout the latest stable version (v1.18.0).
安装
您可以从 PyPI 或从源代码安装 TRL:
PyPI
使用 pip 或 uv 安装该库:
uv
pip
uv 是一个基于 Rust 的快速 Python 包和项目管理器。有关安装说明,请参阅 安装。
uv pip install trl
pip install trl
源代码
您也可以从源代码安装最新版本。首先克隆仓库,然后使用 pip 运行安装:
git clone https://github.com/huggingface/trl.git
cd trl/
pip install -e .如果您想要开发安装,可以用以下内容替换 pip install:
pip install -e ".[dev]"