rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use best match API tokenizers for Text Completion sources

Browse Source
This commit is contained in:
Cohee
2023-12-02 19:42:15 +02:00
parent c98b23ee72
commit 0477f6a553
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
public/scripts/tokenizers.js
View File

@ -4,7 +4,7 @@ import { chat_completion_sources, model_list, oai_settings } from "./openai.js";
import { groups, selected_group } from "./group-chats.js"; import { groups, selected_group } from "./group-chats.js";
import { getStringHash } from "./utils.js"; import { getStringHash } from "./utils.js";
import { kai_flags } from "./kai-settings.js"; import { kai_flags } from "./kai-settings.js";
import { isKoboldCpp, isMancer, isTabby, textgenerationwebui_settings } from "./textgen-settings.js"; import { isKoboldCpp, isMancer, isOoba, isTabby, textgenerationwebui_settings } from "./textgen-settings.js";
export const CHARACTERS_PER_TOKEN_RATIO = 3.35; export const CHARACTERS_PER_TOKEN_RATIO = 3.35;
const TOKENIZER_WARNING_KEY = 'tokenizationWarningShown'; const TOKENIZER_WARNING_KEY = 'tokenizationWarningShown';
@ -127,8 +127,18 @@ export function getTokenizerBestMatch(forApi) {
// - API must be connected // - API must be connected
// - Kobold must pass a version check // - Kobold must pass a version check
// - Tokenizer haven't reported an error previously // - Tokenizer haven't reported an error previously
if (kai_flags.can_use_tokenization && !sessionStorage.getItem(TOKENIZER_WARNING_KEY) && online_status !== 'no_connection') { const hasTokenizerError = sessionStorage.getItem(TOKENIZER_WARNING_KEY);
return tokenizers.API; const isConnected = online_status !== 'no_connection';
const isTokenizerSupported = isOoba() || isTabby() || isKoboldCpp();
if (!hasTokenizerError && isConnected) {
if (forApi === 'kobold' && kai_flags.can_use_tokenization) {
return tokenizers.API;
}
if (forApi === 'textgenerationwebui' && isTokenizerSupported) {
return tokenizers.API;
}
} }
return tokenizers.LLAMA; return tokenizers.LLAMA;