runtime error

Exit code: 1. Reason: /huggingface.co/AIDC-AI/Ovis2-1B: - configuration_ovis.py - modeling_aimv2.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. modeling_ovis.py: 0%| | 0.00/28.4k [00:00<?, ?B/s] modeling_ovis.py: 100%|██████████| 28.4k/28.4k [00:00<00:00, 78.1MB/s] A new version of the following files was downloaded from https://huggingface.co/AIDC-AI/Ovis2-1B: - modeling_ovis.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. model.safetensors: 0%| | 0.00/2.53G [00:00<?, ?B/s] model.safetensors: 0%| | 3.51M/2.53G [00:01<20:20, 2.07MB/s] model.safetensors: 3%|▎ | 69.9M/2.53G [00:04<02:16, 18.0MB/s] model.safetensors: 21%|██▏ | 539M/2.53G [00:05<00:15, 126MB/s]  model.safetensors: 52%|█████▏ | 1.33G/2.53G [00:06<00:04, 284MB/s] model.safetensors: 92%|█████████▏| 2.33G/2.53G [00:07<00:00, 470MB/s] model.safetensors: 100%|██████████| 2.53G/2.53G [00:08<00:00, 310MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 19, in <module> model = AutoModelForCausalLM.from_pretrained(model_name, File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 559, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4097, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/home/user/.cache/huggingface/modules/transformers_modules/AIDC-AI/Ovis2-1B/642d0807bf09b89ed52ea97230ceab26c9a6ed2d/modeling_ovis.py", line 293, in __init__ version.parse(importlib.metadata.version("flash_attn")) >= version.parse("2.6.3")), \ AssertionError: Using `flash_attention_2` requires having `flash_attn>=2.6.3` installed.

Container logs: