pytorch

09d5c181 - disable attn_mask for mha because crash