[refactor] transformer tokenizer

This commit is contained in:
kwaroran
2023-06-07 09:23:29 +09:00
parent b631da7e72
commit 9974ff0df1
2 changed files with 4 additions and 10 deletions

View File

@@ -6,7 +6,7 @@ import { tokenizeTransformers } from "./transformers/transformer";
async function encode(data:string):Promise<(number[]|Uint32Array)>{
let db = get(DataBase)
if(db.aiModel === 'novellist'){
return await tokenizeTransformers('trin',data)
return await tokenizeTransformers('naclbit/trin_tokenizer_v3',data)
}
return await tikJS(data)
}