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"); }