forked from mirrors/gecko-dev
84 lines
2.5 KiB
Diff
84 lines
2.5 KiB
Diff
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
|
|
|