Fixmask (#2538)

* revert mask
OpenNMT · Dec 18, 2023 · 2509d93 · 2509d93
1 parent 23f7916
commit 2509d93
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/onmt/modules/multi_headed_attn.py b/onmt/modules/multi_headed_attn.py
@@ -469,7 +469,6 @@ def forward(
                         (key_pad_mask, y), 2
                     )
                     key_pad_mask = self.layer_cache[1]["key_pad_mask"]
-                    mask = key_pad_mask.unsqueeze(1)
         else:
             # Retrieve keys and values from linear layers (training mode).
             key = self.maybe_ckpt(self.linear_keys, key)