Update gigaam_transformers.py

gigaam_transformers.py

@@ -432,7 +432,7 @@ class GigaAMRNNTHF(PreTrainedModel):
                     last_label = torch.tensor([[hyp[-1]]]).to(x.device)
                     new_symbols += 1
 
-        return torch.tensor([hyp], dtype=torch.int32)
+        return torch.tensor(hyp, dtype=torch.int32)
 
     @torch.inference_mode()
     def generate(self, input_features: Tensor, input_lengths: Tensor, **kwargs) -> torch.Tensor: