openai~=0.27.0
setuptools
termcolor~=1.0
diskcache
joblib~=1.1
tiktoken~=0.4

[huggingface]
transformers>=4.31.0
torch>=2.0
accelerate>=0.21.0

[ort]
transformers>=4.31.0
torch>=2.0
accelerate>=0.21.0
onnx>=1.14.0
datasets>=2.14.4
evaluate>=0.4.0
xformers>=0.0.20
peft>=0.4.0
optimum[onnxruntime]>=1.11.0

[ort-gpu]
transformers>=4.31.0
torch>=2.0
accelerate>=0.21.0
onnx>=1.14.0
datasets>=2.14.4
evaluate>=0.4.0
xformers>=0.0.20
peft>=0.4.0
optimum[onnxruntime-gpu]>=1.11.0
