diffusers

af279434 - Flax tests: don't hardcode number of devices (#1175)

Flax tests: don't hardcode number of devices (#1175) Flax tests: don't hardcode number of devices. This makes it possible to test on CPU/GPU. However, expected slices are only checked when there are 8 devices.