Browse Source

fix to return only the text token ids

Jong Wook Kim 2 năm trước cách đây
mục cha
commit
02aa851a49
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      whisper/transcribe.py

+ 1 - 1
whisper/transcribe.py

@@ -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,