speechbrain
/

SLU-direct-SLURP-hubert-enc

Spoken language understanding

Model card Files Files and versions Community

Adel-Moumen commited on Feb 19

Commit

6644fb4

•

1 Parent(s): 4ecdfc4

Update custom_interface.py

Files changed (1) hide show

custom_interface.py +1 -1

custom_interface.py CHANGED Viewed

@@ -86,7 +86,7 @@ class CustomSLUDecoder(Pretrained):
         with torch.no_grad():
             wavs, wav_lens = wavs.to(self.device), wav_lens.to(self.device)
             encoder_out = self.encode_batch(wavs)
-            predicted_tokens, scores = self.mods.beam_searcher(
                 encoder_out, wav_lens
             )
             predicted_words = [

         with torch.no_grad():
             wavs, wav_lens = wavs.to(self.device), wav_lens.to(self.device)
             encoder_out = self.encode_batch(wavs)
+            predicted_tokens, scores, _, _ = self.mods.beam_searcher(
                 encoder_out, wav_lens
             )
             predicted_words = [