Aphrodite tokenizer

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

View File

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