runtime error

Exit code: 1. Reason: config.json: 0%| | 0.00/1.39k [00:00<?, ?B/s] config.json: 100%|██████████| 1.39k/1.39k [00:00<00:00, 12.1MB/s] pytorch_model.bin: 0%| | 0.00/312M [00:00<?, ?B/s] pytorch_model.bin: 100%|█████████▉| 312M/312M [00:00<00:00, 329MB/s] model.safetensors: 0%| | 0.00/312M [00:00<?, ?B/s] model.safetensors: 100%|█████████▉| 312M/312M [00:00<00:00, 315MB/s] generation_config.json: 0%| | 0.00/293 [00:00<?, ?B/s] generation_config.json: 100%|██████████| 293/293 [00:00<00:00, 2.42MB/s] tokenizer_config.json: 0%| | 0.00/44.0 [00:00<?, ?B/s] tokenizer_config.json: 100%|██████████| 44.0/44.0 [00:00<00:00, 243kB/s] source.spm: 0%| | 0.00/842k [00:00<?, ?B/s] source.spm: 100%|██████████| 842k/842k [00:00<00:00, 30.4MB/s] target.spm: 0%| | 0.00/813k [00:00<?, ?B/s] target.spm: 100%|██████████| 813k/813k [00:00<00:00, 23.9MB/s] vocab.json: 0%| | 0.00/1.72M [00:00<?, ?B/s] vocab.json: 100%|██████████| 1.72M/1.72M [00:00<00:00, 48.2MB/s] Device set to use cpu Traceback (most recent call last): File "/home/user/app/app.py", line 44, 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 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: