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

[saba1999AI]

Traceback (most recent call last):
File "/content/LatentSync/gradio_app.py", line 3, in
from scripts.inference import main
File "/content/LatentSync/scripts/inference.py", line 18, in
from diffusers import AutoencoderKL, DDIMScheduler
File "/usr/local/lib/python3.11/dist-packages/diffusers/init.py", line 36, in
from .models import (
File "/usr/local/lib/python3.11/dist-packages/diffusers/models/init.py", line 33, in
from .controlnet_flax import FlaxControlNetModel
File "/usr/local/lib/python3.11/dist-packages/diffusers/models/controlnet_flax.py", line 25, in
from .modeling_flax_utils import FlaxModelMixin
File "/usr/local/lib/python3.11/dist-packages/diffusers/models/modeling_flax_utils.py", line 45, in
class FlaxModelMixin:
File "/usr/local/lib/python3.11/dist-packages/diffusers/models/modeling_flax_utils.py", line 192, in FlaxModelMixin
def init_weights(self, rng: jax.random.KeyArray) -> Dict:
^^^^^^^^^^^^^^^^^^^
AttributeError: module 'jax.random' has no attribute 'KeyArray'

[productdeployer]

here is the fix

https://www.youtube.com/watch?v=XlaUP0PrQ9U