Stable diffusion: AttributeError: module ‘jax.random‘ has no attribu ...

错误提示

安装
from diffusers import StableDiffusionPipeline 提示错误
   AttributeError: module ‘jax.random’ has no attribute ‘KeyArray’
  办理方法

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

复制代码

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

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

复制代码

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

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。