STATE.hasNode set in worker

Mattk70 · Nov 8, 2024 · 09c5e8e · 09c5e8e
1 parent 6b676ed
commit 09c5e8e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/js/worker.js b/js/worker.js
@@ -642,8 +642,12 @@ async function spawnListWorker() {
             if (message === 'list-model-ready') {
                 return resolve(worker);
             } else if (message === "tfjs-node") {
-                event.data.available || (STATE.detect.backend = 'webgpu');
-                UI.postMessage({event: 'tfjs-node', hasNode: event.data.available})
+                STATE.hasNode = true;
+                if (!event.data.available) {
+                    STATE.detect.backend = 'webgpu';
+                    STATE.hasNode = false;
+                }
+                UI.postMessage({event: 'tfjs-node', hasNode: STATE.hasNode})
             }
         };