optimum
a39b1f56 - Add encoder-decoder model export and inference (#851)

Commit

1 year ago

Add encoder-decoder model export and inference (#851) * add encoder decoder model * update tests * update docs and tests * fixed tests * update tests * update tests * update tests * update tests * update tests * update tests * Update tests/onnxruntime/test_modeling.py Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> * Apply suggestions from code review Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> * udpate testt * change seq2seq-lm to text-generation * fix task * fic tests --------- Co-authored-by: Mohit Sharma <mohit@huggingface.co> Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com>

References

#851 - Add encoder decoder model

Author

mht-sharma

Parents

b87f8a69

Files11

docs/source/exporters/onnx
- overview.mdx
optimum
- exporters
  - onnx
    - config.py
    - model_configs.py
  - tasks.py
- onnxruntime
  - modeling_seq2seq.py
- utils
  - normalized_config.py
tests
- exporters
  - exporters_utils.py
  - onnx
    - test_exporters_onnx_cli.py
    - test_onnx_export.py
- onnxruntime
  - test_modeling.py
  - utils_onnxruntime_tests.py

optimum a39b1f56 - Add encoder-decoder model export and inference (#851)

optimum
a39b1f56 - Add encoder-decoder model export and inference (#851)