PyPI page
Home page
Author:
OpenRLHF Team
Summary:
A Ray-based High-performance RLHF framework.
Latest version:
0.10.4
Required dependencies:
accelerate
|
aiohttp
|
bitsandbytes
|
datasets
|
deepspeed
|
einops
|
flash-attn
|
grpcio
|
huggingface-hub
|
isort
|
jsonlines
|
loralib
|
optimum
|
optree
|
packaging
|
peft
|
pylatexenc
|
pynvml
|
ray
|
sympy
|
tensorboard
|
torch
|
torchdata
|
torchmetrics
|
tqdm
|
transformers
|
transformers-stream-generator
|
wandb
|
wheel
Optional dependencies:
liger-kernel
|
ring-flash-attn
|
vllm
Downloads last day:
130
Downloads last week:
435
Downloads last month:
6,163