Spilum.Net/mtf-nlp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use CPU for the transformer

Browse Source
This commit is contained in:
Apher 2026-07-08 00:43:01 +00:00
parent a5a0096fdc
commit c1552c5b18
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/server.js
View File

@ -70,7 +70,8 @@ function cosineSimilarity(a, b) {
async function initModel() { async function initModel() {
extractor = await pipeline("feature-extraction", `${config.NLP_TRANSFORMER}`, { extractor = await pipeline("feature-extraction", `${config.NLP_TRANSFORMER}`, {
dtype: 'fp32', dtype: 'fp32',
quantized: false quantized: false,
device: "cpu",
}); });
console.log("MiniLM model loaded"); console.log("MiniLM model loaded");
} }