Fix the padding bug in BannedSequenceProcessor (facebookresearch#141)

vb000 · Nov 6, 2023 · 6197f6e · 6197f6e
1 parent fa7f9c4
commit 6197f6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/fairseq2/generation/step_processor.py b/src/fairseq2/generation/step_processor.py
@@ -106,7 +106,7 @@ def __call__(self, seqs: Tensor, probs: Tensor, lprob: bool = False) -> None:
 
         if (len_delta := banned_prefix_len - seqs.size(1)) > 0:
             # (N, S) -> (N, S_pre)
-            seqs = pad(seqs, (len_delta, -1))
+            seqs = pad(seqs, (len_delta, 0), value=-1)
         elif len_delta < 0:
             # (N, S) -> (N, S_pre)
             seqs = seqs[:, -banned_prefix_len:]