onnxruntime/onnxruntime/python/tools/transformers/models/llama/requirements.txt
ironman 9acaf534a6
Benchmark - Updating llama-2 requirement files (#19716)
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
2024-03-04 07:29:58 -08:00

7 lines
No EOL
112 B
Text

optimum>=1.14.1
transformers>=4.33.2,<= 4.37.2
torch>=2.2.0
onnx>=1.14.0
datasets>=2.8.0
protobuf==3.20.2
psutil