openai~=1.14.0
diskcache~=5.6.3
joblib~=1.3.2
tiktoken~=0.5.1
ratemate~=0.1
humanize~=4.8.0

[accelerate]
accelerate~=0.23.0

[dev]
black~=23.10.0
ruff~=0.1.1
pytest~=7.4.2
pytest-cov~=4.1.0

[flash-attn]
flash_attn~=2.3.2

[hf]
torch~=2.1.0
transformers~=4.34.1

[ort]
optimum[onnxruntime]~=1.13.2

[ort-gpu]
optimum[onnxruntime-gpu]~=1.13.2

[quant]
bitsandbytes~=0.41.1

[xformers]
xformers~=0.0.22
