Error

[astrotically]

ran everything and got this error:

Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/import_utils.py", line 704, in _get_module
return importlib.import_module("." + module_name, self.name)
File "/usr/lib/python3.10/importlib/init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "", line 1050, in _gcd_import
File "", line 1027, in _find_and_load
File "", line 1006, in _find_and_load_unlocked
File "", line 688, in _load_unlocked
File "", line 883, in exec_module
File "", line 241, in _call_with_frames_removed
File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/stable_diffusion/init.py", line 95, in
from ...schedulers.scheduling_pndm_flax import PNDMSchedulerState
File "/usr/local/lib/python3.10/dist-packages/diffusers/schedulers/scheduling_pndm_flax.py", line 20, in
import flax
File "/usr/local/lib/python3.10/dist-packages/flax/init.py", line 19, in
from . import linen as linen
File "/usr/local/lib/python3.10/dist-packages/flax/linen/init.py", line 47, in
from .attention import (
File "/usr/local/lib/python3.10/dist-packages/flax/linen/attention.py", line 22, in
from flax.linen.linear import default_kernel_init
File "/usr/local/lib/python3.10/dist-packages/flax/linen/linear.py", line 30, in
from jax import ShapedArray
ImportError: cannot import name 'ShapedArray' from 'jax' (/usr/local/lib/python3.10/dist-packages/jax/init.py)

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/content/dalle-playground/backend/app.py", line 63, in
stable_diff_model = StableDiffusionWrapper()
File "/content/dalle-playground/backend/stable_diffusion_wrapper.py", line 7, in init
pipe = DiffusionPipeline.from_pretrained(
File "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
return fn(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/pipeline_utils.py", line 1111, in from_pretrained
cached_folder = cls.download(
File "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
return fn(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/pipeline_utils.py", line 1783, in download
pipeline_class = _get_pipeline_class(
File "/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/pipeline_utils.py", line 401, in _get_pipeline_class
pipeline_cls = getattr(diffusers_module, class_name)
File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/import_utils.py", line 695, in getattr
value = getattr(module, name)
File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/import_utils.py", line 694, in getattr
module = self._get_module(self._class_to_module[name])
File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/import_utils.py", line 706, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.pipelines.stable_diffusion because of the following error (look up to see its traceback):
cannot import name 'ShapedArray' from 'jax' (/usr/local/lib/python3.10/dist-packages/jax/init.py)

help asap!