From cb176a186ee6dab55f55c55562664bef583e3fd6 Mon Sep 17 00:00:00 2001 From: Thomas Mol Date: Sat, 29 Apr 2023 19:51:05 +0200 Subject: [PATCH] added num_workers to fix pickling error --- whisperx/asr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/whisperx/asr.py b/whisperx/asr.py index d9cbff01..b23a54dc 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -207,7 +207,7 @@ def stack(items): return final_iterator def transcribe( - self, audio: Union[str, np.ndarray], batch_size=None + self, audio: Union[str, np.ndarray], batch_size=None, num_workers=0 ): if isinstance(audio, str): audio = load_audio(audio) @@ -232,7 +232,7 @@ def data(audio, segments): segments = [] batch_size = batch_size or self._batch_size - for idx, out in enumerate(self.__call__(data(audio, vad_segments), batch_size=batch_size)): + for idx, out in enumerate(self.__call__(data(audio, vad_segments), batch_size=batch_size, num_workers=num_workers)): text = out['text'] if batch_size in [0, 1, None]: text = text[0]