AttributeError: 'Config' object has no attribute 'define_bool_state'

[Ishihara-Masabumi]

When I tried to install musubi-tuner, the following error occured.

 Traceback (most recent call last):
  File "/home/ishihara/musubi-tuner/hv_generate_video.py", line 29, in <module>
    from networks import lora
  File "/home/ishihara/musubi-tuner/networks/lora.py", line 11, in <module>
    from diffusers import AutoencoderKL
  File "<frozen importlib._bootstrap>", line 1229, in _handle_fromlist
  File "/home/ishihara/anaconda3/lib/python3.11/site-packages/diffusers/utils/import_utils.py", line 911, in __getattr__
    value = getattr(module, name)
            ^^^^^^^^^^^^^^^^^^^^^
  File "/home/ishihara/anaconda3/lib/python3.11/site-packages/diffusers/utils/import_utils.py", line 910, in __getattr__
    module = self._get_module(self._class_to_module[name])
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ishihara/anaconda3/lib/python3.11/site-packages/diffusers/utils/import_utils.py", line 922, in _get_module
    raise RuntimeError(
RuntimeError: Failed to import diffusers.models.autoencoders.autoencoder_kl because of the following error (look up to see its traceback):
'Config' object has no attribute 'define_bool_state'

I think the cause will be the compatibility between flax and jax.
Please let me know how to fix it.