openai~=1.52.0
tiktoken~=0.7.0
ratemate~=0.1
humanize~=4.8.0
xxhash~=3.4
numpy~=1.24
packaging>=22.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
