src/transformers/models/gpt_neox/modeling_gpt_neox.py/attention_mask_func
def attention_mask_func(attention_scores, ltor_mask):
    attention_scores.masked_fill_(~ltor_mask, torch.finfo(attention_scores.dtype).min)
    return attention_scores