Update gigaam_transformers.py

gigaam_transformers.py

@@ -6,6 +6,7 @@ import torch.nn as nn
 import torchaudio
 from .encoder import ConformerEncoder
 from torch import Tensor
+from torch.nn.utils.rnn import pad_sequence
 from transformers import Wav2Vec2CTCTokenizer, Wav2Vec2Processor
 from transformers.configuration_utils import PretrainedConfig
 from transformers.feature_extraction_sequence_utils import \
@@ -445,4 +446,4 @@ class GigaAMRNNTHF(PreTrainedModel):
         for i in range(b):
             inseq = encoder_out[i, :, :].unsqueeze(1)
             preds.append(self._greedy_decode(inseq, encoded_lengths[i]))
-        return torch.cat(preds, dim=1)
+        return pad_sequence(preds, batch_first=True, padding_value=self.config.blank_id)