[feat] remove @xenova/transformers

This commit is contained in:
kwaroran
2023-06-27 22:20:15 +09:00
parent 36f182b795
commit ca6327ed2e
3 changed files with 5 additions and 393 deletions

View File

@@ -1,24 +0,0 @@
import type { PreTrainedTokenizer } from "@xenova/transformers"
type transformerLibType = typeof import("@xenova/transformers");
let tokenizer:PreTrainedTokenizer = null
let transformerLib:transformerLibType
let tokenizerType:string = ''
async function loadTransformers() {
if(!transformerLib){
transformerLib = await import('@xenova/transformers')
}
}
export async function tokenizeTransformers(type:string, text:string) {
await loadTransformers()
if(tokenizerType !== type){
const AutoTokenizer = transformerLib.AutoTokenizer
tokenizer = await AutoTokenizer.from_pretrained(type)
tokenizerType = type
}
return tokenizer.encode(text)
}