You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Optimum cli has an option to export "image-to-image" task. (https://huggingface.co/docs/optimum/en/exporters/onnx/overview)
However optimum doesn't support task type "image-to-image" for ONNX runtime. There needs to be an OrtModelForImage2Image for SuperResolution or Denoising or other similar tasktypes.
Motivation
It really is a mess to figure out how onnxruntime works. And I'm really used to the transformers and optimums pipeline features. But it really was a mess to learn the manual io bindings output_shape mismatch and all stuff for gpu inference. I want to pipeline it in a similar way.
Feature request
Optimum cli has an option to export "image-to-image" task. (https://huggingface.co/docs/optimum/en/exporters/onnx/overview)
However optimum doesn't support task type "image-to-image" for ONNX runtime. There needs to be an OrtModelForImage2Image for SuperResolution or Denoising or other similar tasktypes.
Motivation
It really is a mess to figure out how onnxruntime works. And I'm really used to the transformers and optimums pipeline features. But it really was a mess to learn the manual io bindings output_shape mismatch and all stuff for gpu inference. I want to pipeline it in a similar way.
When I tried
It gives
ValueError
However exporting to onnx works
optimum-cli export onnx -m caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr swin2sr-4x-onnx
Your contribution
I've started working on the task and will submit a PR Soon.
The text was updated successfully, but these errors were encountered: