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

fix issue of clip search stuck on pending model download

Browse source
This commit is contained in:
Abhinav 2024-01-12 02:35:18 +05:30
parent d1fa900d4b
commit c239a9631d
1 changed files with 3 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/services/clipService.ts
View file

@ -192,22 +192,13 @@ async function getOnnxImageSession() {
return onnxImageSessionPromise; return onnxImageSessionPromise;
} }
let onnxTextSessionPromise: Promise<any> = null;
let onnxTextSession: any = null; let onnxTextSession: any = null;
async function getOnnxTextSession() { async function getOnnxTextSession() {
if (onnxTextSession) { if (!onnxTextSession) {
return onnxTextSession;
}
if (onnxTextSessionPromise) {
throw Error(CustomErrors.MODEL_DOWNLOAD_PENDING);
}
onnxTextSessionPromise = (async () => {
const clipModelPath = await getClipTextModelPath('onnx'); const clipModelPath = await getClipTextModelPath('onnx');
return createOnnxSession(clipModelPath); onnxTextSession = await createOnnxSession(clipModelPath);
})(); }
onnxTextSession = await onnxTextSessionPromise;
onnxTextSessionPromise = null;
return onnxTextSession; return onnxTextSession;
} }