修复bug

This commit is contained in:
ChasonJiang 2025-12-30 15:30:17 +08:00
parent 51df9f7384
commit 585305968e

View File

@ -707,7 +707,6 @@ class Text2SemanticDecoder(nn.Module):
if idx == 0:
attn_mask = F.pad(attn_mask[:, :, -1].unsqueeze(-2), (0, 1), value=False)
logits = logits[:, :-1]
else:
attn_mask = F.pad(attn_mask, (0, 1), value=False)