Ensure WebGPU enabled onnx runtime is loaded

decoder_worker.js

@@ -2,10 +2,10 @@
  * Decoder Worker - Runs adapter + decoder in a separate thread
  */
 
-importScripts('https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/ort.min.js');
+importScripts('https://cdn.jsdelivr.net/npm/onnxruntime-web@1.21.0/dist/ort.all.min.js');
 
-// Configure ONNX Runtime to find WASM files from CDN
-ort.env.wasm.wasmPaths = 'https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/';
+// Configure ONNX Runtime paths
+ort.env.wasm.wasmPaths = 'https://cdn.jsdelivr.net/npm/onnxruntime-web@1.21.0/dist/';
 
 const MODEL_CACHE_NAME = 'moonshine-models-v1';
 

encoder_worker.js

@@ -2,10 +2,10 @@
  * Encoder Worker - Runs preprocessor + encoder in a separate thread
  */
 
-importScripts('https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/ort.min.js');
+importScripts('https://cdn.jsdelivr.net/npm/onnxruntime-web@1.21.0/dist/ort.all.min.js');
 
-// Configure ONNX Runtime to find WASM files from CDN
-ort.env.wasm.wasmPaths = 'https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/';
+// Configure ONNX Runtime paths
+ort.env.wasm.wasmPaths = 'https://cdn.jsdelivr.net/npm/onnxruntime-web@1.21.0/dist/';
 
 const MODEL_CACHE_NAME = 'moonshine-models-v1';
 
@@ -244,7 +244,23 @@ async function processMessage(e) {
                 const backend = data.backend || 'wasm';
                 const dtype = 'fp32';
 
+                // Check WebGPU availability
+                if (backend === 'webgpu') {
+                    if (typeof navigator !== 'undefined' && navigator.gpu) {
+                        console.log('WebGPU navigator.gpu is available');
+                        const adapter = await navigator.gpu.requestAdapter();
+                        if (adapter) {
+                            console.log('WebGPU adapter obtained:', adapter);
+                        } else {
+                            throw new Error('WebGPU adapter not available');
+                        }
+                    } else {
+                        throw new Error('WebGPU not supported (navigator.gpu is undefined)');
+                    }
+                }
+
                 const sessionOptions = { executionProviders: [backend] };
+                console.log(`Creating sessions with backend: ${backend}`);
 
                 tailLatency = cfg.n_future * cfg.encoder_depth;
 

index.html

@@ -705,7 +705,7 @@
         </div>
     </div>
 
-    <script src="https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/ort.min.js"></script>
+    <script src="https://cdn.jsdelivr.net/npm/onnxruntime-web@1.21.0/dist/ort.all.min.js"></script>
     <script type="module" src="streaming_asr.js"></script>
 </body>
 </html>