diffusers

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu")

#5864

Merged