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

Merged

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

patrickvonplaten merged 1 commit into huggingface:main from hvaara:jax-local-cpu

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe…

4c884b37

hvaara marked this pull request as ready for review 2 years ago

patrickvonplaten merged 2695ba8e into main 2 years ago

hvaara deleted the jax-local-cpu branch 2 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone