From 621b16f47870b0fcbd113a37c589f46cea5a69a3 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Wed, 7 Aug 2024 17:47:15 +0800 Subject: [PATCH] Pin transformer and optimum version (#21650) ### Description ### Motivation and Context To fix whisper test failure --- .../python/tools/transformers/models/whisper/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/python/tools/transformers/models/whisper/requirements.txt b/onnxruntime/python/tools/transformers/models/whisper/requirements.txt index 689b14ea9a684..979f872ac4c5e 100644 --- a/onnxruntime/python/tools/transformers/models/whisper/requirements.txt +++ b/onnxruntime/python/tools/transformers/models/whisper/requirements.txt @@ -1,11 +1,11 @@ torch>=1.13.0 -transformers>=4.24.0 +transformers>=4.24.0,<= 4.42.4 openai-whisper>=20231117 ffmpeg-python datasets soundfile librosa -optimum +optimum<=1.21.2 onnxruntime-extensions>=0.9.0 onnx==1.16.1 protobuf==3.20.2