From 55a4a69491f2e69521fe83bff0d03afd6da0e051 Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Tue, 5 Mar 2024 17:03:25 +0200 Subject: [PATCH] Check for existence of `onnx_env.wasm` before updating wasmPaths (#621) --- src/env.js | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/src/env.js b/src/env.js index e4e6e28..a912dbe 100644 --- a/src/env.js +++ b/src/env.js @@ -53,14 +53,15 @@ const localModelPath = RUNNING_LOCALLY ? path.join(__dirname, DEFAULT_LOCAL_MODEL_PATH) : DEFAULT_LOCAL_MODEL_PATH; -// Set path to wasm files. This is needed when running in a web worker. -// https://onnxruntime.ai/docs/api/js/interfaces/Env.WebAssemblyFlags.html#wasmPaths -// We use remote wasm files by default to make it easier for newer users. -// In practice, users should probably self-host the necessary .wasm files. -onnx_env.wasm.wasmPaths = RUNNING_LOCALLY - ? path.join(__dirname, '/dist/') - : `https://cdn.jsdelivr.net/npm/@xenova/transformers@${VERSION}/dist/`; - +if (onnx_env?.wasm) { + // Set path to wasm files. This is needed when running in a web worker. + // https://onnxruntime.ai/docs/api/js/interfaces/Env.WebAssemblyFlags.html#wasmPaths + // We use remote wasm files by default to make it easier for newer users. + // In practice, users should probably self-host the necessary .wasm files. + onnx_env.wasm.wasmPaths = RUNNING_LOCALLY + ? path.join(__dirname, '/dist/') + : `https://cdn.jsdelivr.net/npm/@xenova/transformers@${VERSION}/dist/`; +} /** * Global variable used to control execution. This provides users a simple way to configure Transformers.js.