From c1552c5b185ea810e29ecf56e9a48cbeb1710b85 Mon Sep 17 00:00:00 2001
From: Apher <apher@spilum.net>
Date: Wed, 8 Jul 2026 00:43:01 +0000
Subject: [PATCH] Use CPU for the transformer

---
 src/server.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/server.js b/src/server.js
index fa11f5a..54f8b9b 100644
--- 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");
 }