[Layers] Increased the threshold for enabling flashAttn #428

Open abenmao opened 1 month ago