Merge pull request #2707 from AlpinDale/aphrodite-tokenizer fix: pass model parameter for aphrodite tokenizer

e1b7660448c2fa40d7f37fe39907135df7d1027f

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed
2 files changed, +3 -1Showing whitespace changes
public/scripts/tokenizers.js+1 -0
@@ -838,6 +838,7 @@ function getTextgenAPITokenizationParams(str) {
838838 url: getTextGenServer(),
839839 legacy_api: textgen_settings.legacy_api && (textgen_settings.type === OOBA || textgen_settings.type === APHRODITE),
840840 vllm_model: textgen_settings.vllm_model,
841+ aphrodite_model: textgen_settings.aphrodite_model,
841842 };
842843}
843844
src/endpoints/tokenizers.js+2 -1
@@ -812,6 +812,7 @@ router.post('/remote/textgenerationwebui/encode', jsonParser, async function (re
812812 const baseUrl = String(request.body.url);
813813 const legacyApi = Boolean(request.body.legacy_api);
814814 const vllmModel = String(request.body.vllm_model) || '';
815+ const aphroditeModel = String(request.body.aphrodite_model) || '';
815816
816817 try {
817818 const args = {
@@ -847,7 +848,7 @@ router.post('/remote/textgenerationwebui/encode', jsonParser, async function (re
847848 break;
848849 case TEXTGEN_TYPES.APHRODITE:
849850 url += '/v1/tokenize';
850851 args.body = JSON.stringify({ 'model': aphroditeModel, 'prompt': text });
851852 break;
852853 default:
853854 url += '/v1/internal/encode';