runtime error

Exit code: 1. Reason: s/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' The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> from diffusers import AutoencoderKL File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 944, in __getattr__ value = getattr(module, name) File "/usr/local/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 943, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 955, 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): module 'torch.library' has no attribute 'custom_op'

Container logs: