diff --git a/package-lock.json b/package-lock.json index ee8f375af..5061c1e25 100644 --- a/package-lock.json +++ b/package-lock.json @@ -40,7 +40,7 @@ "png-chunks-extract": "^1.0.0", "response-time": "^2.3.2", "sanitize-filename": "^1.6.3", - "sillytavern-transformers": "^2.6.1", + "sillytavern-transformers": "^2.6.2", "simple-git": "^3.19.1", "uniqolor": "^1.1.0", "vectra": "^0.2.2", @@ -3768,9 +3768,9 @@ } }, "node_modules/sillytavern-transformers": { - "version": "2.6.1", - "resolved": "https://registry.npmjs.org/sillytavern-transformers/-/sillytavern-transformers-2.6.1.tgz", - "integrity": "sha512-mosQ21oyKDQr8d+qmnaIzUlkDDUhhYrb0kccPBefjedLfSahbrUcakpI68JiBf6CdB65KHUp9P1IVrLjiVICTA==", + "version": "2.6.2", + "resolved": "https://registry.npmjs.org/sillytavern-transformers/-/sillytavern-transformers-2.6.2.tgz", + "integrity": "sha512-C9l7mEjSHpOMuAAXrCY0r4Mv8FKlrgTr3XV51evybtXc1bk38KzWSyMukFOC+dVCsuTdqBCyjhdBeF4+6mrHzQ==", "dependencies": { "onnxruntime-web": "1.14.0" }, diff --git a/package.json b/package.json index 56f2aeb88..07fa4b297 100644 --- a/package.json +++ b/package.json @@ -31,7 +31,7 @@ "png-chunks-extract": "^1.0.0", "response-time": "^2.3.2", "sanitize-filename": "^1.6.3", - "sillytavern-transformers": "^2.6.1", + "sillytavern-transformers": "^2.6.2", "simple-git": "^3.19.1", "uniqolor": "^1.1.0", "vectra": "^0.2.2", diff --git a/src/classify.mjs b/src/classify.mjs index c4be593b4..46d959126 100644 --- a/src/classify.mjs +++ b/src/classify.mjs @@ -1,7 +1,7 @@ import { pipeline, TextClassificationPipeline, env } from 'sillytavern-transformers'; import path from 'path'; -env.backends.onnx = 'onnxruntime-web'; +env.backends.onnx.wasm.numThreads = 1; class PipelineAccessor { /**