saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Whisper] Fix provider in export procress (#16545)

Browse source 
type of parameter `provider` is string.
This commit is contained in:
PeixuanZuo 2023-07-10 11:55:36 +08:00 committed by GitHub
parent cb4bf4f5c8
commit 2f56815344
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py
View file

@ -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")