diffusers

[Bug fix] Fix batch size attention head size mismatch

#3214

Merged