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.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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