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

Aphrodite tokenizer

Browse Source
This commit is contained in:
Cohee
2024-02-02 01:09:03 +02:00
parent 4b845dd442
commit 10d78f9a25
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
public/scripts/tokenizers.js
View File

@ -35,7 +35,7 @@ export const SENTENCEPIECE_TOKENIZERS = [
//tokenizers.NERD2, //tokenizers.NERD2,
]; ];
export const TEXTGEN_TOKENIZERS = [OOBA, TABBY, KOBOLDCPP, LLAMACPP]; export const TEXTGEN_TOKENIZERS = [OOBA, TABBY, KOBOLDCPP, LLAMACPP, APHRODITE];
const TOKENIZER_URLS = { const TOKENIZER_URLS = {
[tokenizers.GPT2]: { [tokenizers.GPT2]: {

4
src/endpoints/tokenizers.js
View File

@ -628,6 +628,10 @@ router.post('/remote/textgenerationwebui/encode', jsonParser, async function (re
url += '/tokenize'; url += '/tokenize';
args.body = JSON.stringify({ 'content': text }); args.body = JSON.stringify({ 'content': text });
break; break;
case TEXTGEN_TYPES.APHRODITE:
url += '/v1/tokenize';
args.body = JSON.stringify({ 'prompt': text });
break;
default: default:
url += '/v1/internal/encode'; url += '/v1/internal/encode';
args.body = JSON.stringify({ 'text': text }); args.body = JSON.stringify({ 'text': text });