transformers

4294f0c3 - Llama: partial 4d masks (#29731)

Llama: partial 4d masks (#29731) * partial 4d masks * Apply suggestions from code review Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>