-
Notifications
You must be signed in to change notification settings - Fork 3.4k
Open
Labels
ep:WebGPUort-web webgpu providerort-web webgpu providermodel:transformerissues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc.issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc.platform:webissues related to ONNX Runtime web; typically submitted using templateissues related to ONNX Runtime web; typically submitted using template
Description
Describe the issue
When attempting to run https://huggingface.co/onnx-community/rtdetr_r50vd on WebGPU, I receive the following error:
To reproduce
JSFiddle: https://jsfiddle.net/j8uc3k91/
const url = 'https://huggingface.co/onnx-community/rtdetr_r18vd_coco_o365/resolve/main/onnx/model.onnx?download=true';
const session = await ort.InferenceSession.create(url, {
executionProviders: ['webgpu'],
});
const dims = [1, 3, 640, 640];
const input = new ort.Tensor(
'float32',
new Float32Array(dims.reduce((acc, x) => acc * x, 1)),
dims,
)
// prepare feeds. use model input names as keys.
const feeds = { pixel_values: input };
// feed inputs and run
const results = await session.run(feeds);
console.log(results)
Urgency
Blocks RT-DETR use in Transformers.js w/ WebGPU
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
1.18.0
Execution Provider
'webgpu' (WebGPU)
Metadata
Metadata
Assignees
Labels
ep:WebGPUort-web webgpu providerort-web webgpu providermodel:transformerissues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc.issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc.platform:webissues related to ONNX Runtime web; typically submitted using templateissues related to ONNX Runtime web; typically submitted using template