Browse Source

Fixed CoW RuntimeError in DecodingTask.run() (#240)

Corentin Jemine 2 years ago
parent
commit
9e653bd0ea
1 changed files with 1 additions and 1 deletions
  1. 1 1
      whisper/decoding.py

+ 1 - 1
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)