From 56edb14e441168d8ad3740c837e183958d95a31a Mon Sep 17 00:00:00 2001 From: kwaroran Date: Fri, 5 Jan 2024 23:54:01 +0900 Subject: [PATCH] Update transformers import and add config parameter to runTransformers function --- src/ts/process/embedding/transformers.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/ts/process/embedding/transformers.ts b/src/ts/process/embedding/transformers.ts index b3706bb2..858604f9 100644 --- a/src/ts/process/embedding/transformers.ts +++ b/src/ts/process/embedding/transformers.ts @@ -1,4 +1,4 @@ -import transformers, { AutoTokenizer, Pipeline, pipeline, type DataArray, type SummarizationOutput } from '@xenova/transformers'; +import transformers, { AutoTokenizer, pipeline, type SummarizationOutput } from '@xenova/transformers'; transformers.env.localModelPath = "https://sv.risuai.xyz/transformers/" @@ -25,10 +25,10 @@ type TransformersBodyType = { } -export const runTransformers = async (baseText:string, model:string,bodyTemplate:TransformersBodyType) => { +export const runTransformers = async (baseText:string, model:string,config:transformers.TextGenerationConfig = {}) => { let text = baseText let generator = await pipeline('text-generation', model); - let output = await generator(text) as transformers.TextGenerationOutput + let output = await generator(text, config) as transformers.TextGenerationOutput const outputOne = output[0] return outputOne }