From 180dcefe40130beca5492e423120f7db92ae79c5 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 9 Sep 2023 16:55:54 +0300
Subject: [PATCH] Patch onnx to always use wasm

---
 package-lock.json | 8 ++++----
 package.json      | 2 +-
 src/classify.mjs  | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

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