Public
  1. Public

onnxruntime

Public

Commits

Tianlei WuGitHub
26546f81fe8
Tianlei Wu authored and GitHub committed 268d2283c04
Export GPT-2 ONNX model without postion_ids and attention_mask inputs (#4852)

* Export GPT-2 ONNX model without postion_ids and attention_mask inputs
* allow benchmark_gpt2 on user's model
* refactor:  get_dummy_inputs returns a data class.