Преглед изворни кода

fix to return only the text token ids

Jong Wook Kim пре 2 година
родитељ
комит
02aa851a49
1 измењених фајлова са 1 додато и 1 уклоњено
  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,