diff --git a/whisper_live/server.py b/whisper_live/server.py index be0e959..3305911 100644 --- a/whisper_live/server.py +++ b/whisper_live/server.py @@ -397,6 +397,7 @@ class ServeClientTensorRT(ServeClientBase): language=self.language, task=self.task ) + self.warmup() # threading self.trans_thread = threading.Thread(target=self.speech_to_text) @@ -410,6 +411,12 @@ 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") + for i in range(warmup_steps): + last_segment = self.transcriber.transcribe(mel) def set_eos(self, eos): self.lock.acquire()