runtime error

Exit code: 1. Reason: rror: 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 45, in <module> vlm_model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner-v2").to(device).eval() File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4097, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/models/paligemma/modeling_paligemma.py", line 302, in __init__ self.vision_tower = AutoModel.from_config(config=config.vision_config) File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 439, in from_config model_class = _get_model_class(config, cls._model_mapping) File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 388, in _get_model_class supported_models = model_mapping[type(config)] File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 763, in __getitem__ return self._load_attr_from_module(model_type, model_name) File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 777, in _load_attr_from_module return getattribute_from_module(self._modules[module_name], attr) File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 693, in getattribute_from_module if hasattr(module, attr): 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.siglip.modeling_siglip because of the following error (look up to see its traceback): No module named 'flash_attn_2_cuda'

Container logs: