稳定扩散：AttributeError：模块“jax.random”没有属性“KeyArray”

Question

当我在colab上运行稳定扩散时https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/stable_diffusion.ipynb
没有任何修改，

上线失败

from diffusers import StableDiffusionPipeline 

错误是

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

我该如何解决这个问题或任何线索？

导入应该可以工作，ipynb 应该可以正常运行。

Answer 1

最后，我们需要降级jax， 尝试从最新到早期的每一个，幸运的是它适用于

jax==0.4.23 jaxlib==0.4.23