onnxruntime
9f6d452c - Fix `ValueError` when testing PyTorch performance (#14450)

Commit
3 years ago
Fix `ValueError` when testing PyTorch performance (#14450) ### Description Fixed an exception that is thrown inside `transformers` when trying to test PyTorch performance: ``` > python convert_generation.py -m gpt2 --output gpt2_greedy_search.onnx --num_beams 1 --num_return_sequences 1 --torch_performance
Author
Parents
Loading