Procházet zdrojové kódy

fix to return only the text token ids

Jong Wook Kim před 2 roky
rodič
revize
02aa851a49
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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,