Bring Flax attention naming in sync with PyTorch#2511

pcuenca merged 1 commit intomainhuggingface/diffusers:mainfrom

flax-naminghuggingface/diffusers:flax-namingCopy head branch name to clipboard