Merge remote-tracking branch 'upstream/main' into save_transcript

2024-02-01 12:14:09 +05:30
parent f4027de343 e64bc9f3d6
commit f590446865
6 changed files with 15 additions and 10 deletions
@@ -414,10 +414,10 @@ class ServeClientTensorRT(ServeClientBase):

    def warmup(self, warmup_steps=10):
        logging.info("[INFO:] Warming up TensorRT engine..")
-        mel, duration = self.transcriber.log_mel_spectrogram("tests/jfk.flac")
+        mel, _ = self.transcriber.log_mel_spectrogram("tests/jfk.flac")
        for i in range(warmup_steps):
-            last_segment = self.transcriber.transcribe(mel)
-
+            self.transcriber.transcribe(mel)
+    
    def set_eos(self, eos):
        self.lock.acquire()
        self.eos = eos