fix to return only the text token ids

openai · Nov 16, 2022 · 02aa851 · 02aa851
1 parent 76148a5
commit 02aa851
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/whisper/transcribe.py b/whisper/transcribe.py
@@ -157,7 +157,7 @@ def add_segment(
                 "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,