megatron.model.utils.attention_mask_func# megatron.model.utils.attention_mask_func(attention_scores, attention_mask)#