Remove TransformersBodyType
This commit is contained in:
@@ -2,29 +2,6 @@ import transformers, { AutoTokenizer, pipeline, type SummarizationOutput } from
|
|||||||
|
|
||||||
transformers.env.localModelPath = "https://sv.risuai.xyz/transformers/"
|
transformers.env.localModelPath = "https://sv.risuai.xyz/transformers/"
|
||||||
|
|
||||||
type TransformersBodyType = {
|
|
||||||
max_new_tokens: number,
|
|
||||||
do_sample: boolean,
|
|
||||||
temperature: number,
|
|
||||||
top_p: number,
|
|
||||||
typical_p: number,
|
|
||||||
repetition_penalty: number,
|
|
||||||
encoder_repetition_penalty: number,
|
|
||||||
top_k: number,
|
|
||||||
min_length: number,
|
|
||||||
no_repeat_ngram_size: number,
|
|
||||||
num_beams: number,
|
|
||||||
penalty_alpha: number,
|
|
||||||
length_penalty: number,
|
|
||||||
early_stopping: boolean,
|
|
||||||
truncation_length: number,
|
|
||||||
ban_eos_token: boolean,
|
|
||||||
stopping_strings: number,
|
|
||||||
seed: number,
|
|
||||||
add_bos_token: boolean,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
export const runTransformers = async (baseText:string, model:string,config:transformers.TextGenerationConfig = {}) => {
|
export const runTransformers = async (baseText:string, model:string,config:transformers.TextGenerationConfig = {}) => {
|
||||||
let text = baseText
|
let text = baseText
|
||||||
let generator = await pipeline('text-generation', model);
|
let generator = await pipeline('text-generation', model);
|
||||||
|
|||||||
Reference in New Issue
Block a user