fix: logprobs for Aphrodite

Since Aphrodite follows the same logprobs style as ooba, we can enable it easily.
This commit is contained in:
AlpinDale 2024-01-26 07:15:39 +00:00 committed by GitHub
parent 3cf01e765a
commit 8af3939f09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions

View File

@ -712,6 +712,7 @@ function parseTextgenLogprobs(token, logprobs) {
}
switch (settings.type) {
case APHRODITE:
case OOBA: {
/** @type {Record<string, number>[]} */
const topLogprobs = logprobs.top_logprobs;