Skip to content

Commit

Permalink
Benchmark - Updating llama-2 requirement files (microsoft#19716)
Browse files Browse the repository at this point in the history
### 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. -->
  • Loading branch information
ironmangithub authored Mar 4, 2024
1 parent 9460597 commit 9acaf53
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Original file line number Diff line number Diff line change
@@ -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
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 9acaf53

Please sign in to comment.