diffusers

add reshape to fix use_memory_efficient_attention in flax

#7918

Merged