Fix state_bag parameter in MHA pos encoder (#46)

facebookresearch · Sep 15, 2023 · 5301e0a · 5301e0a
1 parent 7e6b7e2
commit 5301e0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/fairseq2/nn/transformer/multihead_attention.py b/src/fairseq2/nn/transformer/multihead_attention.py
@@ -457,7 +457,7 @@ def forward(
         v = v.flatten(0, 1)
 
         if self.pos_encoder is not None:
-            q = self.pos_encoder(q, padding_mask, state_bag)
+            q = self.pos_encoder(q, padding_mask, state_bag=state_bag)
             k = self.pos_encoder(k, key_padding_mask)
 
         mask_pad = 0