rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Patch onnx to always use wasm

This commit is contained in:
Cohee 2023-09-09 16:55:54 +03:00
parent 307e666c27
commit 180dcefe40
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
package-lock.json generated
View File

@ -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"
},

2
package.json
View File

@ -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",

2
src/classify.mjs
View File

@ -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 {
/**