Skip to content

GitLab

Unverified Commit 2059b8d9 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

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

parent 8aaf3d53
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 9 deletions
tests/models/test_internvl.py
View file @ 2059b8d9
...@@ -3,7 +3,6 @@ from typing import List, Optional, Tuple, Type ...@@ -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({ ...@@ -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",
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment