Add best match tokenizer logic for DeepSeek and Nemo

This commit is contained in:
Cohee
2025-01-02 23:22:53 +02:00
parent 2d4bae17cc
commit 40b05bb6b1

View File

@ -325,6 +325,12 @@ export function getTokenizerBestMatch(forApi) {
if (model.includes('gemma')) {
return tokenizers.GEMMA;
}
if (model.includes('nemo') || model.includes('pixtral')) {
return tokenizers.NEMO;
}
if (model.includes('deepseek')) {
return tokenizers.DEEPSEEK;
}
if (model.includes('yi')) {
return tokenizers.YI;
}