load audio for warmup

2024-01-29 12:15:39 +05:30
parent 4d94c6b38b
commit 30d2cffb93
1 changed files with 2 additions and 2 deletions
@@ -11,7 +11,7 @@ import numpy as np
 from whisper.tokenizer import get_tokenizer
 from whisper_live.tensorrt_utils import (mel_filters, store_transcripts,
                           write_error_stats, load_audio_wav_format,
-                           pad_or_trim)
+                           pad_or_trim, load_audio)

 import tensorrt_llm
 import tensorrt_llm.logger as logger
@@ -337,4 +337,4 @@ def decode_wav_file(
    if normalizer:
        prediction = normalizer(prediction)
    
-    return prediction.strip()
+    return prediction.strip()