Skip to content

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

Description

@amrzv

Hello.
When running google colab notebook this error is raised on the line from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler

The issue is that jax library needs to be downgraded to 0.4.23, so adding the following into the first cell resolves the error

jax==0.4.23 jaxlib==0.4.23

stackoverflow

Should the requirements.txt be patched? I can create an MR.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions