add the webgpu backend

Mattk70 · Mar 9, 2024 · 74a98e0 · 74a98e0
1 parent 840ccb8
commit 74a98e0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/index.html b/index.html
@@ -181,6 +181,10 @@ <h5 class="offcanvas-title" id="offcanvasExampleLabel">Settings</h5>
                                     <input class="form-check-input" type="radio" name="backend" id="webgl" value="webgl">
                                     <label class="form-check-label" for="webgl">GPU</label>
                                 </div>
+                                <div class="form-check form-check-inline">
+                                    <input class="form-check-input" type="radio" name="backend" id="webgpu" value="webgpu">
+                                    <label class="form-check-label" for="webgpu">webGPU</label>
+                                </div>
                             </div>
                         </div>
                     </div>

diff --git a/js/model.js b/js/model.js
@@ -1,4 +1,5 @@
 const tf = require('@tensorflow/tfjs-node');
+require('@tensorflow/tfjs-backend-webgpu');
 const fs = require('node:fs');
 const path = require('node:path');
 let DEBUG = false;

diff --git a/package.json b/package.json
@@ -194,6 +194,7 @@
     "@fast-csv/format": "^4.3.5",
     "@popperjs/core": "^2.9.2",
     "@tensorflow/tfjs-core": "4.17.0",
+    "@tensorflow/tfjs-backend-webgpu": "4.17.0",
     "@tensorflow/tfjs-node": "4.17.0",
     "bootstrap": "5.2.3",
     "browser-id3-writer": "^4.4.0",