runtime error

Exit code: 1. Reason: e "<frozen importlib._bootstrap_external>", line 883, in exec_module File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed File "/usr/local/lib/python3.10/site-packages/transformers/models/whisper/modeling_whisper.py", line 52, in <module> from ...modeling_flash_attention_utils import _flash_attention_forward File "/usr/local/lib/python3.10/site-packages/transformers/modeling_flash_attention_utils.py", line 27, in <module> from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input # noqa File "/usr/local/lib/python3.10/site-packages/flash_attn/__init__.py", line 3, in <module> from flash_attn.flash_attn_interface import ( File "/usr/local/lib/python3.10/site-packages/flash_attn/flash_attn_interface.py", line 15, in <module> import flash_attn_2_cuda as flash_attn_gpu ModuleNotFoundError: No module named 'flash_attn_2_cuda' The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 28, in <module> pipe = pipeline( File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/__init__.py", line 926, in pipeline framework, model = infer_framework_load_model( File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/base.py", line 260, in infer_framework_load_model _class = getattr(transformers_module, architecture, None) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1767, in __getattr__ value = getattr(module, name) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1766, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1780, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.models.whisper.modeling_whisper because of the following error (look up to see its traceback): No module named 'flash_attn_2_cuda'

Container logs: