torch>=2.3.0
transformers>=4.45.0
accelerate>=0.33.0
peft>=0.12.0
bitsandbytes>=0.43.0
pillow>=10.0
datasets>=2.16.0
huggingface-hub>=0.24.0
tqdm>=4.66.0
einops>=0.8.0
safetensors>=0.4.3
numpy>=1.24
sentencepiece>=0.2.0
trl>=0.12.0
wandb>=0.17.0
tensorboard>=2.16.0

[all]
openllava[cli,cuda,dev,eval,rl,serve,tpu,ui]

[cli]
click>=8.0
typer>=0.12
rich>=13.0

[cuda]
nvidia-cuda-runtime-cu12
nvidia-cuda-nvcc-cu12
nvidia-cublas-cu12

[dev]
pytest>=8.0
black>=24.0
ruff>=0.4
pre-commit>=3.6

[eval]
lmms-eval>=0.2.0

[rl]
trl>=0.12.0
vllm>=0.5.0

[serve]
fastapi>=0.110
uvicorn[standard]>=0.29

[tpu]
jax>=0.4.20
jaxlib>=0.4.20
flax>=0.8.0
torch_xla>=2.3.0

[ui]
gradio>=4.0
