beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix multiple createOnnxSession called issue

Browse source
This commit is contained in:
Abhinav 2024-01-11 10:21:47 +05:30
parent 40ab33a903
commit 37a125d153
1 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/services/clipService.ts
View file

@ -180,24 +180,28 @@ async function createOnnxSession(modelPath: string) {
});
}
let onnxImageSession: any = null;
let onnxImageSessionPromise: Promise<any> = null;
async function getOnnxImageSession() {
if (!onnxImageSession) {
if (!onnxImageSessionPromise) {
onnxImageSessionPromise = (async () => {
const clipModelPath = await getClipImageModelPath('onnx');
onnxImageSession = createOnnxSession(clipModelPath);
return createOnnxSession(clipModelPath);
})();
}
return onnxImageSession;
return onnxImageSessionPromise;
}
let onnxTextSession: any = null;
let onnxTextSessionPromise: Promise<any> = null;
async function getOnnxTextSession() {
if (!onnxTextSession) {
if (!onnxTextSessionPromise) {
onnxTextSessionPromise = (async () => {
const clipModelPath = await getClipTextModelPath('onnx');
onnxTextSession = createOnnxSession(clipModelPath);
onnxTextSessionPromise = createOnnxSession(clipModelPath);
})();
}
return onnxTextSession;
return onnxTextSessionPromise;
}
let tokenizer: Tokenizer = null;