CyberZHG's picture
Use input attention mask instead of casual mask in attention
333bf48