From 9523e5c1f9721c2ed246adf5a6fb501da43875d7 Mon Sep 17 00:00:00 2001
From: Manan17 <shahmanan170602@gmail.com>
Date: Tue, 10 Mar 2026 21:12:24 +0000
Subject: [PATCH] fixing embedding model search

---
 studio/frontend/src/hooks/use-hf-model-search.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/studio/frontend/src/hooks/use-hf-model-search.ts b/studio/frontend/src/hooks/use-hf-model-search.ts
index 5f603a71e8..330455e819 100644
--- a/studio/frontend/src/hooks/use-hf-model-search.ts
+++ b/studio/frontend/src/hooks/use-hf-model-search.ts
@@ -25,6 +25,13 @@ const EXCLUDED_TAGS = new Set([
   "ctranslate2",
 ]);
 
+// Embedding / sentence-transformer models ship with onnx/openvino as additional
+// export formats — they should not be excluded by the tag check above.
+const EMBEDDING_TAGS = new Set([
+  "sentence-transformers",
+  "feature-extraction",
+]);
+
 function withPopularitySort(
   input: Parameters<typeof fetch>[0],
   init?: Parameters<typeof fetch>[1],
@@ -56,7 +63,8 @@ function makeMapModel(excludeGguf: boolean) {
       safetensors?: { total: number };
       tags?: string[];
     };
-    if (m.tags?.some((t) => EXCLUDED_TAGS.has(t))) {
+    const isEmbedding = m.tags?.some((t) => EMBEDDING_TAGS.has(t));
+    if (!isEmbedding && m.tags?.some((t) => EXCLUDED_TAGS.has(t))) {
       return null;
     }
     if (excludeGguf && m.tags?.includes("gguf")) {