diffusers

Flax tests: don't hardcode number of devices

#1175

Merged