瀏覽代碼

fix issue of clip search stuck on pending model download

Abhinav 1 年之前
父節點
當前提交
c239a9631d
共有 1 個文件被更改,包括 3 次插入12 次删除
  1. 3 12
      src/services/clipService.ts

+ 3 - 12
src/services/clipService.ts

@@ -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) {
-        return onnxTextSession;
-    }
-    if (onnxTextSessionPromise) {
-        throw Error(CustomErrors.MODEL_DOWNLOAD_PENDING);
-    }
-    onnxTextSessionPromise = (async () => {
+    if (!onnxTextSession) {
         const clipModelPath = await getClipTextModelPath('onnx');
         const clipModelPath = await getClipTextModelPath('onnx');
-        return createOnnxSession(clipModelPath);
-    })();
-    onnxTextSession = await onnxTextSessionPromise;
-    onnxTextSessionPromise = null;
+        onnxTextSession = await createOnnxSession(clipModelPath);
+    }
     return onnxTextSession;
     return onnxTextSession;
 }
 }