optimum
c3914e90 - Support for custom input shapes in exporters onnx (#575)

Commit

2 years ago

Support for custom input shapes in exporters onnx (#575) * add support for custom input shapes in exporters onnx * add tests * fix test * fix bug * Update tests/exporters/test_onnx_export.py Co-authored-by: regisss <15324346+regisss@users.noreply.github.com> * remove run slow * fix _get_models_to_test * fixup post merge * yet fixup post merge * use f string * authorize unused arguments in input generators * use hf-internal-testing models * remove redundant argument * fixup merge * remove redundant arg * fix messed merge * change blenderbot * workign deit * valid gptj * valid marian * add whisper * fix test * fix marian * nit * fix bloom Co-authored-by: regisss <15324346+regisss@users.noreply.github.com>

References

#575 - Support for custom input shapes in exporters onnx

Author

fxmarty

Parents

178728b2

Files19

.github/workflows
- test_exporters.yml
optimum
- commands/export
  - onnx.py
- exporters
  - onnx
    - __main__.py
    - base.py
    - config.py
    - convert.py
    - model_configs.py
    - utils.py
  - tasks.py
- onnxruntime
  - modeling_decoder.py
  - modeling_ort.py
  - modeling_seq2seq.py
  - trainer.py
  - trainer_seq2seq.py
- utils
  - input_generators.py
  - testing_utils.py
tests/exporters
- exporters_utils.py
- test_onnx_config_loss.py
- test_onnx_export.py

optimum c3914e90 - Support for custom input shapes in exporters onnx (#575)

optimum
c3914e90 - Support for custom input shapes in exporters onnx (#575)