Stable diffusion: AttributeError: module ‘jax.random‘ has no attribute ‘KeyArray‘

错误提示

安装
from diffusers import StableDiffusionPipeline 提示错误

AttributeError: module ‘jax.random’ has no attribute ‘KeyArray’

解决方法

pip install "jax[cuda12_pip]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

或者
pip install "jax[cuda12_local]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

在colab里面，可以运行以下命令行

!pip install "jax[cuda12_local]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
!pip install diffusers==0.11.1
!pip install transformers scipy ftfy accelerate

参考：
https://stackoverflow.com/questions/78302031/stable-diffusion-attributeerror-module-jax-random-has-no-attribute-keyarray

文章来源于互联网:Stable diffusion: AttributeError: module ‘jax.random‘ has no attribute ‘KeyArray‘