runtime error

Exit code: 1. Reason: in /usr/local/lib/python3.10/site-packages (from torch->flash-attn) (3.5.1) Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.10/site-packages (from sympy>=1.13.3->torch->flash-attn) (1.3.0) Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/site-packages (from jinja2->torch->flash-attn) (3.0.3) Building wheels for collected packages: flash-attn Building wheel for flash-attn (pyproject.toml): started Building wheel for flash-attn (pyproject.toml): finished with status 'done' Created wheel for flash-attn: filename=flash_attn-2.8.3-py3-none-any.whl size=372051 sha256=44cfce1b4d99e34dd0c342ea2908c50c8c12e8b35ae97e5d6a737c1f9a567854 Stored in directory: /home/user/.cache/pip/wheels/f5/05/1e/a6726e9eee2e7ee6151dbfed113e89d220dd3964ba617ab32d Successfully built flash-attn Installing collected packages: flash-attn Successfully installed flash-attn-2.8.3 Traceback (most recent call last): File "/home/user/app/app.py", line 16, in <module> from main import load_v2 File "/home/user/app/main.py", line 6, in <module> from MeshAnything.models.meshanything_v2 import MeshAnythingV2 File "/home/user/app/MeshAnything/models/meshanything_v2.py", line 7, in <module> from MeshAnything.models.shape_opt import ShapeOPTConfig File "/home/user/app/MeshAnything/models/shape_opt.py", line 2, in <module> from transformers.models.opt.modeling_opt import OPTForCausalLM, OPTModel, OPTDecoder, OPTLearnedPositionalEmbedding, OPTDecoderLayer File "/usr/local/lib/python3.10/site-packages/transformers/models/opt/modeling_opt.py", line 46, in <module> from flash_attn import flash_attn_func, flash_attn_varlen_func 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'

Container logs: