Tienson Qin 6 месяцев назад
Родитель
Сommit
e8d6ae1a2f

+ 3 - 2
src/main/frontend/inference_worker/text_embedding.cljs

@@ -177,8 +177,9 @@
 (def available-embedding-models
   {"Xenova/all-MiniLM-L6-v2" {:tf-config {:dtype "fp32"}
                               :hnsw-config {:dims 384}}
-   "onnx-community/Qwen3-Embedding-0.6B-ONNX" {:tf-config {:dtype "fp16"}
-                                               :hnsw-config {:dims 1024}}})
+   ;; "onnx-community/Qwen3-Embedding-0.6B-ONNX" {:tf-config {:dtype "fp16"}
+   ;;                                             :hnsw-config {:dims 1024}}
+   })
 
 (def ^:private *load-model-progress (atom nil))
 

+ 9 - 7
src/main/frontend/worker/embedding.cljs

@@ -135,13 +135,15 @@
       (m/? (task--update-index-info!* repo infer-worker)))))
 
 (defn- get-partition-size
-  [repo]
-  (let [conn (worker-state/get-datascript-conn repo)
-        embedding-model-name (ldb/get-key-value @conn :logseq.kv/graph-text-embedding-model-name)]
-    (case embedding-model-name
-      "onnx-community/Qwen3-Embedding-0.6B-ONNX"
-      100
-      500)))
+  [_repo]
+  500
+  ;; (let [conn (worker-state/get-datascript-conn repo)
+  ;;       embedding-model-name (ldb/get-key-value @conn :logseq.kv/graph-text-embedding-model-name)]
+  ;;   (case embedding-model-name
+  ;;     "onnx-community/Qwen3-Embedding-0.6B-ONNX"
+  ;;     100
+  ;;     500))
+  )
 
 (defn- task--embedding-stale-blocks!
   "embedding outdated block-data