From 3df109adfccedeb134dea4ba2ea9a2da89872048 Mon Sep 17 00:00:00 2001
From: Isuxiz Slidder <48672727+Isuxiz@users.noreply.github.com>
Date: 星期一, 31 三月 2025 17:51:52 +0800
Subject: [PATCH] Update model.py to fix "IndexError: index 1 is out of bounds for dimension 1 with size 0" (#2454)
---
funasr/models/transformer/utils/vgg2l.py | 8 ++------
1 files changed, 2 insertions(+), 6 deletions(-)
diff --git a/funasr/models/transformer/utils/vgg2l.py b/funasr/models/transformer/utils/vgg2l.py
index fb45d0b..7466bac 100644
--- a/funasr/models/transformer/utils/vgg2l.py
+++ b/funasr/models/transformer/utils/vgg2l.py
@@ -39,9 +39,7 @@
else:
self.output = torch.nn.Linear(128 * ((idim // 2) // 2), odim)
- def forward(
- self, feats: torch.Tensor, feats_mask: torch.Tensor
- ) -> Union[
+ def forward(self, feats: torch.Tensor, feats_mask: torch.Tensor) -> Union[
Tuple[torch.Tensor, torch.Tensor],
Tuple[Tuple[torch.Tensor, torch.Tensor], torch.Tensor],
]:
@@ -62,9 +60,7 @@
b, c, t, f = vgg_output.size()
- vgg_output = self.output(
- vgg_output.transpose(1, 2).contiguous().view(b, t, c * f)
- )
+ vgg_output = self.output(vgg_output.transpose(1, 2).contiguous().view(b, t, c * f))
if feats_mask is not None:
vgg_mask = self.create_new_mask(feats_mask)
--
Gitblit v1.9.1