|
561 | 561 | get_stop_token_ids=lambda tok: tok.convert_tokens_to_ids(['<|im_end|>', '<|endoftext|>']), # noqa: E501
|
562 | 562 | hf_output_post_proc=model_utils.minicpmv_trunc_hf_output,
|
563 | 563 | patch_hf_runner=model_utils.minicpmo_26_patch_hf_runner,
|
564 |
| - # FIXME: https://huggingface.co/openbmb/MiniCPM-V-2_6/discussions/55 |
| 564 | + # FIXME: https://huggingface.co/openbmb/MiniCPM-o-2_6/discussions/49 |
565 | 565 | marks=[pytest.mark.skip("HF import fails")],
|
566 | 566 | ),
|
567 | 567 | "minicpmv_26": VLMTestInfo(
|
|
574 | 574 | get_stop_token_ids=lambda tok: tok.convert_tokens_to_ids(['<|im_end|>', '<|endoftext|>']), # noqa: E501
|
575 | 575 | hf_output_post_proc=model_utils.minicpmv_trunc_hf_output,
|
576 | 576 | patch_hf_runner=model_utils.minicpmv_26_patch_hf_runner,
|
577 |
| - # FIXME: https://huggingface.co/openbmb/MiniCPM-V-2_6/discussions/55 |
578 |
| - marks=[pytest.mark.skip("HF import fails")], |
579 | 577 | ),
|
580 | 578 | "minimax_vl_01": VLMTestInfo(
|
581 | 579 | models=["MiniMaxAI/MiniMax-VL-01"],
|
|
611 | 609 | patch_hf_runner=model_utils.ovis_patch_hf_runner,
|
612 | 610 | marks=[large_gpu_mark(min_gb=32)],
|
613 | 611 | ),
|
614 |
| - "ovis1_6": VLMTestInfo( |
615 |
| - models=["AIDC-AI/Ovis1.6-Llama3.2-3B"], |
616 |
| - test_type=(VLMTestType.IMAGE, VLMTestType.MULTI_IMAGE), |
617 |
| - prompt_formatter=lambda img_prompt: f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nYou are a helpful and honest multimodal assistant.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{img_prompt}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n", # noqa: E501 |
618 |
| - img_idx_to_prompt=lambda idx: "<image>\n", # noqa: E501 |
619 |
| - max_model_len=4096, |
620 |
| - max_num_seqs=2, |
621 |
| - dtype="half", |
622 |
| - # use sdpa mode for hf runner since ovis2 didn't work with flash_attn |
623 |
| - hf_model_kwargs={"llm_attn_implementation": "sdpa"}, |
624 |
| - patch_hf_runner=model_utils.ovis_patch_hf_runner, |
625 |
| - ), |
626 | 612 | "ovis2": VLMTestInfo(
|
627 | 613 | models=["AIDC-AI/Ovis2-1B"],
|
628 | 614 | test_type=(VLMTestType.IMAGE, VLMTestType.MULTI_IMAGE),
|
|
0 commit comments