diff --git a/index.html b/index.html index b31dc980..921dfd04 100644 --- a/index.html +++ b/index.html @@ -181,6 +181,10 @@
Settings
+
+ + +
diff --git a/js/model.js b/js/model.js index 3ec95f88..4b8d57c6 100644 --- 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 index 1193a7fa..f7884449 100644 --- 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",