search

Uminosachi / sd-webui-inpaint-anything

Inpaint Anything extension performs stable diffusion inpainting on a browser UI using masks from Segment Anything.
Apache License 2.0
1.11k stars 101 forks source link

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

Open Silentou opened 5 months ago

Silentou commented 5 months ago

Error loading script: inpaint_anything.py Traceback (most recent call last): File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/scripts.py", line 508, in load_scripts script_module = script_loading.load_module(scriptfile.path) File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/script_loading.py", line 13, in load_module module_spec.loader.exec_module(module) File "", line 883, in exec_module File "", line 241, in _call_with_frames_removed File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/extensions/sd-webui-inpaint-anything/scripts/inpaint_anything.py", line 17, in from diffusers import (DDIMScheduler, EulerAncestralDiscreteScheduler, EulerDiscreteScheduler, File "/usr/local/lib/python3.10/dist-packages/diffusers/init.py", line 28, in from .pipelines import OnnxRuntimeModel File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/init.py", line 19, in from .dance_diffusion import DanceDiffusionPipeline File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/dance_diffusion/init.py", line 1, in from .pipeline_dance_diffusion import DanceDiffusionPipeline File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py", line 21, in from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/pipeline_utils.py", line 39, in from ..models.modeling_utils import _LOW_CPU_MEM_USAGE_DEFAULT File "/usr/local/lib/python3.10/dist-packages/diffusers/models/init.py", line 33, in from .controlnet_flax import FlaxControlNetModel File "/usr/local/lib/python3.10/dist-packages/diffusers/models/controlnet_flax.py", line 25, in from .modeling_flax_utils import FlaxModelMixin File "/usr/local/lib/python3.10/dist-packages/diffusers/models/modeling_flax_utils.py", line 45, in class FlaxModelMixin: File "/usr/local/lib/python3.10/dist-packages/diffusers/models/modeling_flax_utils.py", line 192, in FlaxModelMixin def init_weights(self, rng: jax.random.KeyArray) -> Dict: File "/usr/local/lib/python3.10/dist-packages/jax/_src/deprecations.py", line 54, in getattr raise AttributeError(f"module {module!r} has no attribute {name!r}") AttributeError: module 'jax.random' has no attribute 'KeyArray'