torch>=1.4.0
transformers>=4.18.0
numpy>=1.18.2
accelerate
datasets
tyro>=0.5.7

[benchmark]
wandb
ghapi
openrlbenchmark==0.2.1a5
requests
deepspeed

[deepspeed]
deepspeed>=0.9.5

[dev]
parameterized
pytest
pytest-xdist
accelerate
peft>=0.4.0
diffusers>=0.18.0
deepspeed>=0.9.5
wandb
ghapi
openrlbenchmark==0.2.1a5
requests
deepspeed
bitsandbytes>=0.41.0

[diffusers]
diffusers>=0.18.0

[peft]
peft>=0.4.0

[quantization]
bitsandbytes>=0.41.0

[test]
parameterized
pytest
pytest-xdist
accelerate
peft>=0.4.0
diffusers>=0.18.0
