From 9e653bd0ea0f1e9493cb4939733e9de249493cfb Mon Sep 17 00:00:00 2001 From: Corentin Jemine Date: Tue, 4 Oct 2022 17:49:31 +0200 Subject: [PATCH] Fixed CoW RuntimeError in DecodingTask.run() (#240) --- whisper/decoding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisper/decoding.py b/whisper/decoding.py index c604631..ed8d900 100644 --- a/whisper/decoding.py +++ b/whisper/decoding.py @@ -615,7 +615,7 @@ class DecodingTask: n_audio: int = mel.shape[0] audio_features: Tensor = self._get_audio_features(mel) # encoder forward pass - tokens: Tensor = torch.tensor([self.initial_tokens]).expand(n_audio, -1) + tokens: Tensor = torch.tensor([self.initial_tokens]).repeat(n_audio, 1) # detect language if requested, overwriting the language token languages, language_probs = self._detect_language(audio_features, tokens)