Explorar el Código

fix to return only the text token ids

Jong Wook Kim hace 2 años
padre
commit
02aa851a49
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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,