embedding worker fix
This commit is contained in:
@@ -3,9 +3,9 @@ import { parentPort } from 'worker_threads';
|
||||
|
||||
let embedder: any;
|
||||
|
||||
parentPort?.on('message', async ({ id, text, model, modelDir }) => {
|
||||
parentPort?.on('message', async ({text, model, modelDir }) => {
|
||||
if(!embedder) embedder = await pipeline('feature-extraction', 'Xenova/' + model, {quantized: true, cache_dir: modelDir});
|
||||
const output = await embedder(text, { pooling: 'mean', normalize: true });
|
||||
const embedding = Array.from(output.data);
|
||||
parentPort?.postMessage({ id, embedding });
|
||||
parentPort?.postMessage({embedding});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user