runtime error

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.5.9.post1-py3-none-any.whl size=120889689 sha256=5022ba11d48bf74926da9c16260f4ea2b9bb7f4e29bdb4bd6e1383ad1c55d16f Stored in directory: /home/user/.cache/pip/wheels/cc/ad/f6/7ccf0238790d6346e9fe622923a76ec218e890d356b9a2754a Successfully built flash-attn Installing collected packages: einops, flash-attn Successfully installed einops-0.8.0 flash-attn-2.5.9.post1 A new version of the following files was downloaded from https://huggingface.co/HuggingFaceM4/Florence-2-DocVQA: - configuration_florence2.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/HuggingFaceM4/Florence-2-DocVQA: - modeling_florence2.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 20, in <module> 'mihirinamdar/Florence-2-FT-DocVQA': AutoModelForCausalLM.from_pretrained('mihirinamdar/Florence-2-FT-DocVQA', trust_remote_code=True).to("cuda").eval(), 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 3710, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/home/user/.cache/huggingface/modules/transformers_modules/HuggingFaceM4/Florence-2-DocVQA/c757833c585c682e2a57f70e124f9f8128f6049f/modeling_florence2.py", line 2532, in __init__ assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now' AssertionError: only DaViT is supported for now

Container logs: