PyPI page
Home page
Author:
OpenRLHF Team
Summary:
A Ray-based High-performance RLHF framework.
Latest version:
0.10.3
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:
125
Downloads last week:
902
Downloads last month:
4,238