runtime error

Exit code: 1. Reason: , 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 33, in <module> from gradio_app.gradio_3dgen import create_ui as create_3d_ui File "/home/user/app/gradio_app/gradio_3dgen.py", line 7, in <module> from gradio_app.custom_models.mvimg_prediction import run_mvprediction File "/home/user/app/gradio_app/custom_models/mvimg_prediction.py", line 8, in <module> from gradio_app.custom_models.utils import load_pipeline File "/home/user/app/gradio_app/custom_models/utils.py", line 6, in <module> from custum_3d_diffusion import modules File "/home/user/app/custum_3d_diffusion/modules.py", line 14, in <module> from custum_3d_diffusion.trainings import base, image2mvimage_trainer, image2image_trainer File "/home/user/app/custum_3d_diffusion/trainings/base.py", line 14, in <module> from custum_3d_diffusion.custum_modules.unifield_processor import ConfigurableUNet2DConditionModel, AttnConfig File "/home/user/app/custum_3d_diffusion/custum_modules/unifield_processor.py", line 3, in <module> from diffusers import UNet2DConditionModel 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.unets.unet_2d_condition because of the following error (look up to see its traceback): module 'torch.library' has no attribute 'custom_op'

Container logs: