From 8af3939f09f63b22d2184064010a4d7ebe67fd36 Mon Sep 17 00:00:00 2001 From: AlpinDale <52078762+AlpinDale@users.noreply.github.com> Date: Fri, 26 Jan 2024 07:15:39 +0000 Subject: [PATCH] fix: logprobs for Aphrodite Since Aphrodite follows the same logprobs style as ooba, we can enable it easily. --- public/scripts/textgen-settings.js | 1 + 1 file changed, 1 insertion(+) diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js index 3dbf91590..1d1fe57ae 100644 --- a/public/scripts/textgen-settings.js +++ b/public/scripts/textgen-settings.js @@ -712,6 +712,7 @@ function parseTextgenLogprobs(token, logprobs) { } switch (settings.type) { + case APHRODITE: case OOBA: { /** @type {Record[]} */ const topLogprobs = logprobs.top_logprobs;