diff --git a/onnxruntime/python/tools/transformers/models/llama/requirements-cuda.txt b/onnxruntime/python/tools/transformers/models/llama/requirements-cuda.txt index acd9c23aa4..307afbc122 100644 --- a/onnxruntime/python/tools/transformers/models/llama/requirements-cuda.txt +++ b/onnxruntime/python/tools/transformers/models/llama/requirements-cuda.txt @@ -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 \ No newline at end of file diff --git a/onnxruntime/python/tools/transformers/models/llama/requirements.txt b/onnxruntime/python/tools/transformers/models/llama/requirements.txt index 8b57279295..e991c2f27a 100644 --- a/onnxruntime/python/tools/transformers/models/llama/requirements.txt +++ b/onnxruntime/python/tools/transformers/models/llama/requirements.txt @@ -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 \ No newline at end of file diff --git a/onnxruntime/python/tools/transformers/models/whisper/requirements.txt b/onnxruntime/python/tools/transformers/models/whisper/requirements.txt index 956922dc83..9bbe0d7380 100644 --- a/onnxruntime/python/tools/transformers/models/whisper/requirements.txt +++ b/onnxruntime/python/tools/transformers/models/whisper/requirements.txt @@ -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