From 8ee3da1d8743246a13782cadb38ad24f3ac63972 Mon Sep 17 00:00:00 2001 From: Olga Gerasimova Date: Mon, 14 Nov 2022 18:40:30 +0100 Subject: [PATCH] use merge masks from functional --- torch/nn/modules/transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/nn/modules/transformer.py b/torch/nn/modules/transformer.py index 37e8823edf2c..6d70c51772f9 100644 --- a/torch/nn/modules/transformer.py +++ b/torch/nn/modules/transformer.py @@ -500,7 +500,7 @@ def forward(self, src: Tensor, src_mask: Optional[Tensor] = None, "input/output projection weights or biases requires_grad") if not why_not_sparsity_fast_path: - merged_mask, mask_type = self.self_attn.merge_masks(src_mask, src_key_padding_mask, src) + merged_mask, mask_type = F.merge_masks(src_mask, src_key_padding_mask, src) return torch._transformer_encoder_layer_fwd( src, self.self_attn.embed_dim,