Geekphysique comments

Repositories
Issues
Comments

Results 2 comments of


                                            Geekphysique

AttributeError: module 'jax.random' has no attribute 'KeyArray'

Also seeing the same. AttributeError: module 'jax.random' has no attribute 'KeyArray'

AttributeError: module 'jax.random' has no attribute 'KeyArray'

> jax.random.KeyArray was removed in JAX v0.4.24. current diffusers only work with JAX v0.4.23 or earlier. > > add under requirements: > > !pip install "jax[cuda12_pip]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html Worked like...