From cb67fca738410abc629f517167593078d13d8908 Mon Sep 17 00:00:00 2001 From: Yulong Wang Date: Wed, 18 Aug 2021 11:18:58 -0700 Subject: [PATCH] [js/web] enable 'use_ort_model_bytes_directly' by default (#8734) --- js/web/lib/wasm/session-options.ts | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/js/web/lib/wasm/session-options.ts b/js/web/lib/wasm/session-options.ts index 55780dd8ca..560de879c7 100644 --- a/js/web/lib/wasm/session-options.ts +++ b/js/web/lib/wasm/session-options.ts @@ -33,12 +33,27 @@ const getExecutionMode = (executionMode: 'sequential'|'parallel'): number => { } }; +const appendDefaultOptions = (options: InferenceSession.SessionOptions): void => { + if (!options.extra) { + options.extra = {}; + } + if (!options.extra.session) { + options.extra.session = {}; + } + const session = options.extra.session as Record; + if (!session.use_ort_model_bytes_directly) { + // eslint-disable-next-line camelcase + session.use_ort_model_bytes_directly = '1'; + } +}; + export const setSessionOptions = (options?: InferenceSession.SessionOptions): [number, number[]] => { const wasm = getInstance(); let sessionOptionsHandle = 0; const allocs: number[] = []; const sessionOptions: InferenceSession.SessionOptions = options || {}; + appendDefaultOptions(sessionOptions); try { if (options?.graphOptimizationLevel === undefined) {