heimoshuiyu/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix to return only the text token ids

Browse Source
This commit is contained in:
Jong Wook Kim
2022-11-15 16:25:11 -08:00
parent 76148a56c5
commit 02aa851a49
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/transcribe.py
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,