For people having numpy.dtype size changed error or RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1).