Merge pull request speechbrain#2432 from speechbrain/Adel-Moumen-patch-1

Fix EndToEndSLU
poonehmousavi · Feb 26, 2024 · b26a6d2 · b26a6d2
2 parents 3f9e33a + a4c033c
commit b26a6d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/speechbrain/inference/SLU.py b/speechbrain/inference/SLU.py
@@ -118,7 +118,7 @@ def decode_batch(self, wavs, wav_lens):
  with torch.no_grad():
  wavs, wav_lens = wavs.to(self.device), wav_lens.to(self.device)
  encoder_out = self.encode_batch(wavs, wav_lens)
- predicted_tokens, scores = self.mods.beam_searcher(
+ predicted_tokens, scores, _, _ = self.mods.beam_searcher(
  encoder_out, wav_lens
  )
  predicted_words = [