From 2f56815344bac9a2e009774e701c2139988cfe5f Mon Sep 17 00:00:00 2001 From: PeixuanZuo <94887879+PeixuanZuo@users.noreply.github.com> Date: Mon, 10 Jul 2023 11:55:36 +0800 Subject: [PATCH] [Whisper] Fix provider in export procress (#16545) type of parameter `provider` is string. --- .../python/tools/transformers/models/whisper/convert_to_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py b/onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py index 1b634717ba..eef35db20b 100644 --- a/onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py +++ b/onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py @@ -397,7 +397,7 @@ def main(argv=None): ort_session = create_onnxruntime_session( args.beam_model_output_dir, use_gpu=args.use_gpu, - provider=["CUDAExecutionProvider", "CPUExecutionProvider"] if args.use_gpu else ["CPUExecutionProvider"], + provider=args.provider, ) device = torch.device("cuda:0" if args.use_gpu else "cpu")