From 72bce123a5fc6b32e208086aad3c7025e2428992 Mon Sep 17 00:00:00 2001 From: Manav Rathi Date: Sat, 10 Aug 2024 13:42:26 +0530 Subject: [PATCH] Cleanup --- desktop/src/main/services/ml-worker.ts | 26 +------------------------- 1 file changed, 1 insertion(+), 25 deletions(-) diff --git a/desktop/src/main/services/ml-worker.ts b/desktop/src/main/services/ml-worker.ts index 40c1c5fb5e..68a8eb8349 100644 --- a/desktop/src/main/services/ml-worker.ts +++ b/desktop/src/main/services/ml-worker.ts @@ -201,12 +201,6 @@ const createInferenceSession = async (modelPath: string) => { }); }; -// TODO-ML: Remove me -// const cachedCLIPImageSessionOAI = makeCachedInferenceSession( -// "clip-image-vit-32-float32.onnx", -// 351468764 /* 335 MB */, -// ); - const cachedCLIPImageSession = makeCachedInferenceSession( "mobileclip_s2_image.onnx", 143061211 /* 143 MB */, @@ -229,24 +223,6 @@ export const computeCLIPImageEmbedding = async (input: Float32Array) => { return ensure(results.output).data as Float32Array; }; -// TODO-ML: Remove me -// const cachedCLIPTextSessionOAIQ = makeCachedInferenceSession( -// "clip-text-vit-32-uint8.onnx", -// 64173509 /* 61 MB */, -// ); - -// TODO-ML: Remove me -// const cachedCLIPTextSessionOAI = makeCachedInferenceSession( -// "clip-text-vit-32-float32-int32.onnx", -// 254069585 /* 254 MB */, -// ); - -// TODO-ML: Remove me -// const cachedCLIPTextSession = makeCachedInferenceSession( -// "mobileclip_s2_text.onnx", -// 253895732 /* 253 MB */, -// ); - const cachedCLIPTextSession = makeCachedInferenceSession( "mobileclip_s2_text_int32.onnx", 253895600 /* 253 MB */, @@ -294,7 +270,7 @@ export const computeCLIPTextEmbeddingIfAvailable = async (text: string) => { const cachedFaceDetectionSession = makeCachedInferenceSession( "yolov5s_face_640_640_dynamic.onnx", - 30762872 /* 29.3 MB */, + 30762872 /* 29 MB */, ); /**