diffusers
Flax: add shape argument to `set_timesteps`
#690
Merged

Loading