yizhi-js-lib/ai-box
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

代码优化

Browse Source
This commit is contained in:
Yizhi
2025-03-10 18:12:32 +08:00
parent a966b82963
commit 524dcaecbd
10 changed files with 71 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/backend/mnn/session.ts
View File

@ -1,3 +1,4 @@
import { getConfig } from "../../config";
import { CommonSession, dataTypeFrom, isTypedArray, SessionNodeData, SessionNodeInfo, SessionRunInputOption, SessionRunOutput } from "../common";
export class MNNSession extends CommonSession {
@ -7,7 +8,7 @@ export class MNNSession extends CommonSession {
public constructor(modelData: Uint8Array) {
super();
const addon = require("../../../build/mnn.node")
const addon = require(getConfig("MNN_ADDON_FILE"));
this.#session = new addon.MNNSession(modelData);
}

3
src/backend/ort/session.ts
View File

@ -1,3 +1,4 @@
import { getConfig } from "../../config";
import { CommonSession, dataTypeFrom, isTypedArray, SessionNodeData, SessionNodeInfo, SessionRunInputOption, SessionRunOutput } from "../common";
export class OrtSession extends CommonSession {
@ -7,7 +8,7 @@ export class OrtSession extends CommonSession {
public constructor(modelData: Uint8Array) {
super();
const addon = require("../../../build/ort.node")
const addon = require(getConfig("ORT_ADDON_FILE"));
this.#session = new addon.OrtSession(modelData);
}