[Web] Error: [WebGPU] Kernel "[Conv] /text_encoder/encoder/layers.0/feed_forward/conv_2/Conv" failed. Error: FILTER_IN_CHANNEL should be equal to DATA_CHANNEL

[xenova]

Describe the issue

Attempting to run Xenova/mms-tts-eng on WebGPU produces the following error:

Uncaught Error: [WebGPU] Kernel "[Conv] /text_encoder/encoder/layers.0/feed_forward/conv_2/Conv" failed. Error: FILTER_IN_CHANNEL should be equal to DATA_CHANNEL

To reproduce

1. Download the model from the HF hub: https://huggingface.co/Xenova/mms-tts-eng/resolve/main/onnx/model.onnx?download=true
2. Load the session and set execution provider to webgpu
3. Run the model with attention_mask and input_ids as inputs
4. Observe error.

Note that the WASM backend does not have this issue (the model runs correctly)

Urgency

Blocks vits/mms model usage in Transformers.js

ONNX Runtime Installation

Released Package

ONNX Runtime Version or Commit ID

1.18.0

Execution Provider

'webgpu' (WebGPU)

[xenova]

cc @guschmue

[github-actions]

This issue has been automatically marked as stale due to inactivity and will be closed in 30 days if no further activity occurs. If further support is needed, please provide an update and/or more details.

[xenova]

bump