runtime error

Exit code: 1. Reason: usion File "/usr/local/lib/python3.10/site-packages/hidiffusion/hidiffusion.py", line 13, in <module> from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel File "/usr/local/lib/python3.10/site-packages/diffusers/pipelines/controlnet/multicontrolnet.py", line 1, in <module> from ...models.controlnets.multicontrolnet import MultiControlNetModel File "/usr/local/lib/python3.10/site-packages/diffusers/models/controlnets/__init__.py", line 5, in <module> from .controlnet import ControlNetModel, ControlNetOutput File "/usr/local/lib/python3.10/site-packages/diffusers/models/controlnets/controlnet.py", line 33, in <module> from ..unets.unet_2d_blocks import ( File "/usr/local/lib/python3.10/site-packages/diffusers/models/unets/__init__.py", line 6, in <module> from .unet_2d import UNet2DModel File "/usr/local/lib/python3.10/site-packages/diffusers/models/unets/unet_2d.py", line 24, in <module> from .unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block File "/usr/local/lib/python3.10/site-packages/diffusers/models/unets/unet_2d_blocks.py", line 36, in <module> from ..transformers.dual_transformer_2d import DualTransformer2DModel File "/usr/local/lib/python3.10/site-packages/diffusers/models/transformers/__init__.py", line 20, in <module> from .transformer_chroma import ChromaTransformer2DModel File "/usr/local/lib/python3.10/site-packages/diffusers/models/transformers/transformer_chroma.py", line 33, in <module> from .transformer_flux import FluxAttention, FluxAttnProcessor File "/usr/local/lib/python3.10/site-packages/diffusers/models/transformers/transformer_flux.py", line 29, in <module> from ..attention_dispatch import dispatch_attention_fn File "/usr/local/lib/python3.10/site-packages/diffusers/models/attention_dispatch.py", line 481, in <module> @torch.library.custom_op("flash_attn_3::_flash_attn_forward", mutates_args=(), device_types="cuda") AttributeError: module 'torch.library' has no attribute 'custom_op'

Container logs: