runtime error

Exit code: 1. Reason: p517` option, (possibly combined with `--no-build-isolation`), or adding a `pyproject.toml` file to the source tree of 'flash-attn'. Discussion can be found at https://github.com/pypa/pip/issues/6334 Building wheel for flash-attn (setup.py): started Building wheel for flash-attn (setup.py): finished with status 'done' Created wheel for flash-attn: filename=flash_attn-2.7.4.post1-py3-none-any.whl size=217424 sha256=3e1307185b4dc96183298501ab67e083f02527f873633b0e369a520ab8d3d228 Stored in directory: /home/user/.cache/pip/wheels/59/ce/d5/08ea07bfc16ba218dc65a3a7ef9b6a270530bcbd2cea2ee1ca Successfully built flash-attn Installing collected packages: einops, flash-attn Successfully installed einops-0.8.1 flash-attn-2.7.4.post1 /usr/local/lib/python3.10/site-packages/timm/models/layers/__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning) Traceback (most recent call last): File "/home/user/app/app.py", line 29, in <module> from models.infinity import Infinity File "/home/user/app/models/__init__.py", line 6, in <module> from .infinity import Infinity, sample_with_top_k_top_p_also_inplace_modifying_logits_ File "/home/user/app/models/infinity.py", line 23, in <module> from models.basic import flash_attn_func, flash_fused_op_installed, AdaLNBeforeHead, CrossAttnBlock, SelfAttnBlock, CrossAttention, FastRMSNorm, precompute_rope2d_freqs_grid File "/home/user/app/models/basic.py", line 18, in <module> from flash_attn import flash_attn_func # q, k, or v: BLHc, ret: BLHc 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: