runtime error

Exit code: 1. Reason: -0000(…): 32%|███▏ | 3.15G/9.96G [00:05<00:08, 815MB/s] transformer/diffusion_pytorch_model-0000(…): 51%|█████ | 5.03G/9.96G [00:06<00:04, 1.05GB/s] transformer/diffusion_pytorch_model-0000(…): 67%|██████▋ | 6.64G/9.96G [00:07<00:03, 1.06GB/s] transformer/diffusion_pytorch_model-0000(…): 100%|██████████| 9.96G/9.96G [00:09<00:00, 1.26GB/s] transformer/diffusion_pytorch_model-0000(…): 100%|██████████| 9.96G/9.96G [00:09<00:00, 1.00GB/s] transformer/diffusion_pytorch_model-0000(…): 0%| | 0.00/482M [00:00<?, ?B/s] transformer/diffusion_pytorch_model-0000(…): 42%|████▏ | 201M/482M [00:01<00:02, 115MB/s] transformer/diffusion_pytorch_model-0000(…): 100%|██████████| 482M/482M [00:02<00:00, 235MB/s] (…)ion_pytorch_model.safetensors.index.json: 0%| | 0.00/42.2k [00:00<?, ?B/s] (…)ion_pytorch_model.safetensors.index.json: 100%|██████████| 42.2k/42.2k [00:00<00:00, 126MB/s] config.json: 0%| | 0.00/825 [00:00<?, ?B/s] config.json: 100%|██████████| 825/825 [00:00<00:00, 8.65MB/s] vae/diffusion_pytorch_model.safetensors: 0%| | 0.00/335M [00:00<?, ?B/s] vae/diffusion_pytorch_model.safetensors: 100%|██████████| 335M/335M [00:01<00:00, 277MB/s] vae/diffusion_pytorch_model.safetensors: 100%|██████████| 335M/335M [00:01<00:00, 277MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 37, in <module> import models File "/home/user/app/models/__init__.py", line 1, in <module> from .model import NextDiT_2B_GQA_patch2_Adaln_Refiner, NextDiT_3B_GQA_patch2_Adaln_Refiner, NextDiT_4B_GQA_patch2_Adaln_Refiner, NextDiT_7B_GQA_patch2_Adaln_Refiner File "/home/user/app/models/model.py", line 15, in <module> from flash_attn import flash_attn_varlen_func ModuleNotFoundError: No module named 'flash_attn'

Container logs: