runtime error

Exit code: 1. Reason: in /usr/local/lib/python3.10/site-packages (from sympy->torch->flash-attn) (1.3.0) Building wheels for collected packages: flash-attn 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.8.0.post2-py3-none-any.whl size=286447 sha256=cd5d11b2917e2020e4c8d3f8cdc351f50080ac1202c3567f08b841b28a885b1d Stored in directory: /home/user/.cache/pip/wheels/d3/7f/92/22a4559afd149337bd27c954bab184e10fceda1c77ed331c6f Successfully built flash-attn Installing collected packages: einops, flash-attn Successfully installed einops-0.8.1 flash-attn-2.8.0.post2 [notice] A new release of pip available: 22.3.1 -> 25.1.1 [notice] To update, run: /usr/local/bin/python -m pip install --upgrade pip A new version of the following files was downloaded from https://huggingface.co/HuggingFaceM4/VLM_WebSight_finetuned: - configuration_vmistral.py . Make sure to double-check they do not contain any added malicious code. To avoid downloading new versions of the code file, you can pin a revision. Traceback (most recent call last): File "/home/user/app/app.py", line 25, in <module> MODEL = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 551, in from_pretrained model_class = get_class_from_dynamic_module( File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 502, in get_class_from_dynamic_module final_module = get_cached_module_file( File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 327, in get_cached_module_file modules_needed = check_imports(resolved_module_file) File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 182, in check_imports raise ImportError( ImportError: This modeling file requires the following packages that were not found in your environment: flash_attn. Run `pip install flash_attn`

Container logs: