Use CPU for the transformer

2026-07-08 00:43:01 +00:00 · 2026-07-08 00:43:01 +00:00 · c1552c5b18
commit c1552c5b18
parent a5a0096fdc
1 changed files with 2 additions and 1 deletions
--- a/src/server.js
+++ b/src/server.js
@ -70,7 +70,8 @@ function cosineSimilarity(a, b) {
 async function initModel() {
  extractor = await pipeline("feature-extraction", `${config.NLP_TRANSFORMER}`, {
    dtype: 'fp32',
-    quantized: false
+    quantized: false,
+    device: "cpu",
  });
  console.log("MiniLM model loaded");
 }