Fix for non-streaming

This commit is contained in:
Cohee
2024-02-24 14:50:06 +02:00
parent 445cbda02f
commit 9287ff18de
2 changed files with 7 additions and 1 deletions

View File

@ -764,7 +764,7 @@ async function generateTextGenWithStreaming(generate_data, signal) {
* @param {Object} logprobs - logprobs object returned from the API
* @returns {import('logprobs.js').TokenLogprobs | null} - converted logprobs
*/
function parseTextgenLogprobs(token, logprobs) {
export function parseTextgenLogprobs(token, logprobs) {
if (!logprobs) {
return null;
}