From a96aad6073dacceffbca7765e32c499923a29ab8 Mon Sep 17 00:00:00 2001 From: ThisIsPIRI Date: Fri, 8 Sep 2023 18:31:30 +0900 Subject: [PATCH] Make API tokenization work again for ooba --- server.js | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/server.js b/server.js index 0ae0d38cb..2b0065486 100644 --- a/server.js +++ b/server.js @@ -4005,8 +4005,10 @@ app.post("/tokenize_via_api", jsonParser, async function (request, response) { headers: { "Content-Type": "application/json" } }; - if (main_api == 'textgenerationwebui' && request.body.use_mancer) { - args.headers = Object.assign(args.headers, get_mancer_headers()); + if (main_api == 'textgenerationwebui') { + if (request.body.use_mancer) { + args.headers = Object.assign(args.headers, get_mancer_headers()); + } const data = await postAsync(api_server + "/v1/token-count", args); return response.send({ count: data['results'][0]['tokens'] }); }