xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-21 01:25:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Remove snapshot_download usage in InternVL2 test (#7835)

Browse Source
This commit is contained in:
Isotr0py 2024-08-25 23:53:09 +08:00 committed by GitHub
parent 8aaf3d5347
commit 2059b8d9ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/models/test_internvl.py
View File

@ -3,7 +3,6 @@ from typing import List, Optional, Tuple, Type
import pytest import pytest
import torch import torch
from huggingface_hub import snapshot_download
from PIL.Image import Image from PIL.Image import Image
from transformers import AutoConfig from transformers import AutoConfig
@ -25,17 +24,12 @@ HF_IMAGE_PROMPTS = IMAGE_ASSETS.prompts({
"<|im_start|>User\n<image>\nWhat is the season?<|im_end|>\n<|im_start|>Assistant\n", # noqa: E501 "<|im_start|>User\n<image>\nWhat is the season?<|im_end|>\n<|im_start|>Assistant\n", # noqa: E501
}) })
# we use snapshot_download to prevent conflicts between
# dynamic_module and trust_remote_code for hf_runner
DOWNLOAD_PATTERN = ["*.json", "*.py", "*.safetensors", "*.txt", "*.model"]
models = [ models = [
snapshot_download("OpenGVLab/InternVL2-1B", "OpenGVLab/InternVL2-1B",
allow_patterns=DOWNLOAD_PATTERN), "OpenGVLab/InternVL2-2B",
snapshot_download("OpenGVLab/InternVL2-2B",
allow_patterns=DOWNLOAD_PATTERN),
# Broken due to outdated implementation of Phi-3 # Broken due to outdated implementation of Phi-3
# See: https://huggingface.co/OpenGVLab/InternVL2-4B/discussions/3 # See: https://huggingface.co/OpenGVLab/InternVL2-4B/discussions/3
# snapshot_download("OpenGVLab/InternVL2-4B"), # "OpenGVLab/InternVL2-4B",
] ]