diff options
Diffstat (limited to 'toolkit/components/ml/vendor/gecko.patch')
-rw-r--r-- | toolkit/components/ml/vendor/gecko.patch | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/toolkit/components/ml/vendor/gecko.patch b/toolkit/components/ml/vendor/gecko.patch new file mode 100644 index 0000000000..8c0392e3aa --- /dev/null +++ b/toolkit/components/ml/vendor/gecko.patch @@ -0,0 +1,84 @@ +From 2804004525c65df99cd8a563cfd316e796618ebd Mon Sep 17 00:00:00 2001 +From: =?UTF-8?q?Tarek=20Ziad=C3=A9?= <tarek@ziade.org> +Date: Wed, 10 Apr 2024 10:06:19 +0200 +Subject: [PATCH 1/1] patched +MIME-Version: 1.0 +Content-Type: text/plain; charset=UTF-8 +Content-Transfer-Encoding: 8bit + +Signed-off-by: Tarek Ziadé <tarek@ziade.org> +--- + package.json | 2 +- + src/backends/onnx.js | 8 ++++++-- + tsconfig.json | 13 +++++++++++++ + webpack.config.js | 4 ++++ + 4 files changed, 24 insertions(+), 3 deletions(-) + create mode 100644 tsconfig.json + +diff --git a/package.json b/package.json +index 446fd5f..24825d6 100644 +--- a/package.json ++++ b/package.json +@@ -6,7 +6,7 @@ + "types": "./types/transformers.d.ts", + "type": "module", + "scripts": { +- "typegen": "tsc ./src/transformers.js --allowJs --declaration --emitDeclarationOnly --declarationMap --outDir types", ++ "typegen": "tsc", + "dev": "webpack serve --no-client-overlay", + "build": "webpack && npm run typegen", + "generate-tests": "python -m tests.generate_tests", +diff --git a/src/backends/onnx.js b/src/backends/onnx.js +index 0bee3dc..030eff7 100644 +--- a/src/backends/onnx.js ++++ b/src/backends/onnx.js +@@ -18,8 +18,12 @@ + + // NOTE: Import order matters here. We need to import `onnxruntime-node` before `onnxruntime-web`. + // In either case, we select the default export if it exists, otherwise we use the named export. +-import * as ONNX_NODE from 'onnxruntime-node'; +-import * as ONNX_WEB from 'onnxruntime-web'; ++//import * as ONNX_NODE from 'onnxruntime-node'; ++//import * as ONNX_WEB from 'onnxruntime-web'; ++ ++import 'onnxruntime-web'; ++const ONNX_WEB = ort; ++ + + /** @type {import('onnxruntime-web')} The ONNX runtime module. */ + export let ONNX; +diff --git a/tsconfig.json b/tsconfig.json +new file mode 100644 +index 0000000..8892b18 +--- /dev/null ++++ b/tsconfig.json +@@ -0,0 +1,13 @@ ++{ ++ "files": [ ++ "./src/transformers.js" ++ ], ++ "compilerOptions": { ++ "allowJs": true, ++ "declaration": true, ++ "emitDeclarationOnly": true, ++ "declarationMap": true, ++ "outDir": "types", ++ "types": [] // Or specify only the types you need ++ } ++} +diff --git a/webpack.config.js b/webpack.config.js +index c958b45..8b6b98b 100644 +--- a/webpack.config.js ++++ b/webpack.config.js +@@ -51,4 +51,8 @@ export default { + experiments: { + outputModule: true, + }, ++ externals: { ++ 'onnxruntime-node': 'NOT_USED', ++ 'onnxruntime-web': 'chrome://global/content/ml/ort-dev.js', ++ }, + }; +-- +2.43.0 + |