runtime error

Exit code: 1. Reason: ng_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. A new version of the following files was downloaded from https://huggingface.co/AIDC-AI/Ovis-U1-3B: - configuration_ovis_u1.py - configuration_aimv2.py - modeling_yak.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. Traceback (most recent call last): File "/home/user/app/app.py", line 92, in <module> model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 531, in from_pretrained config, kwargs = AutoConfig.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 1123, in from_pretrained config_class = get_class_from_dynamic_module( File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 570, in get_class_from_dynamic_module return get_class_in_module(class_name, final_module, force_reload=force_download) File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 267, in get_class_in_module module_spec.loader.exec_module(module) File "<frozen importlib._bootstrap_external>", line 883, in exec_module File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed File "/home/user/.cache/huggingface/modules/transformers_modules/AIDC-AI/Ovis-U1-3B/7e9bc1660f8bc8b9050350baee519310a370834a/configuration_ovis_u1.py", line 16, in <module> from .modeling_aimv2 import AIMv2Model File "/home/user/.cache/huggingface/modules/transformers_modules/AIDC-AI/Ovis-U1-3B/7e9bc1660f8bc8b9050350baee519310a370834a/modeling_aimv2.py", line 10, in <module> from flash_attn import flash_attn_varlen_func ImportError: cannot import name 'flash_attn_varlen_func' from 'flash_attn' (unknown location)

Container logs: