fix to return only the text token ids

This commit is contained in:
Jong Wook Kim 2022-11-15 16:25:11 -08:00
parent 76148a56c5
commit 02aa851a49

View File

@ -157,7 +157,7 @@ def transcribe(
"start": start,
"end": end,
"text": text,
"tokens": result.tokens,
"tokens": text_tokens.tolist(),
"temperature": result.temperature,
"avg_logprob": result.avg_logprob,
"compression_ratio": result.compression_ratio,