mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-04 23:59:56 +00:00
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. -->
This commit is contained in:
parent
9460597b21
commit
9acaf534a6
3 changed files with 4 additions and 2 deletions
|
|
@ -2,3 +2,4 @@
|
|||
# Please manually install torch>=2.2.0 with CUDA enabled for the CUDA version installed in your system.
|
||||
# Instructions can be found here: https://pytorch.org/get-started/locally/
|
||||
onnxruntime-gpu>=1.16.2
|
||||
py3nvml
|
||||
|
|
@ -1,6 +1,7 @@
|
|||
optimum>=1.14.1
|
||||
transformers>=4.33.2
|
||||
transformers>=4.33.2,<= 4.37.2
|
||||
torch>=2.2.0
|
||||
onnx>=1.14.0
|
||||
datasets>=2.8.0
|
||||
protobuf==3.20.2
|
||||
psutil
|
||||
|
|
@ -7,8 +7,8 @@ soundfile
|
|||
librosa
|
||||
optimum
|
||||
onnxruntime-extensions>=0.9.0
|
||||
onnx>=1.15.0
|
||||
protobuf==3.20.2
|
||||
numpy==1.23.3
|
||||
onnx>=1.15.0
|
||||
psutil
|
||||
py3nvml
|
||||
|
|
|
|||
Loading…
Reference in a new issue