search

huggingface / optimum

🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
https://huggingface.co/docs/optimum/main/
Apache License 2.0
2.6k stars 477 forks source link

Add test at the ONNX export that axis names are not shared for variable length inputs/outputs #870

Open fxmarty opened 1 year ago

fxmarty commented 1 year ago

Feature request

As per title

Motivation

TensorRT, ONNX Runtime don't like shared axis name with different length.

Would allow to catch issues as https://github.com/huggingface/optimum/pull/871 ahead of time

Your contribution

/

fxmarty commented 1 year ago

Such a test would also have catched https://github.com/huggingface/optimum/issues/882#issuecomment-1470125541

fxmarty commented 1 year ago

Would have catched as well https://github.com/huggingface/optimum/pull/904