Instalation

[pepeballesterostel]

For the installation of diffusers in colab, I am following the colab example. This just to work just fine, but now, after installing diffusers==0.11.1, I cannot import the diffusers library:
AttributeError: module 'jax.random' has no attribute 'KeyArray'

JAX version in colab 0.4.26 is compatible with that of diffusers dependencies: "jax>=0.4.1", "jaxlib>=0.4.1",
Any help to install correctly diffusers in colab is much appreciated!

[tolgacangoz]

Could you just remove ==0.11.1 part, and try with the latest version?