From 2059b8d9caf12072710a7d610dd80954ad7c047e Mon Sep 17 00:00:00 2001 From: Isotr0py <2037008807@qq.com> Date: Sun, 25 Aug 2024 23:53:09 +0800 Subject: [PATCH] [Misc] Remove snapshot_download usage in InternVL2 test (#7835) --- tests/models/test_internvl.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/models/test_internvl.py b/tests/models/test_internvl.py index d032f3be84..243bc857c8 100644 --- a/tests/models/test_internvl.py +++ b/tests/models/test_internvl.py @@ -3,7 +3,6 @@ from typing import List, Optional, Tuple, Type import pytest import torch -from huggingface_hub import snapshot_download from PIL.Image import Image from transformers import AutoConfig @@ -25,17 +24,12 @@ HF_IMAGE_PROMPTS = IMAGE_ASSETS.prompts({ "<|im_start|>User\n\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 = [ - snapshot_download("OpenGVLab/InternVL2-1B", - allow_patterns=DOWNLOAD_PATTERN), - snapshot_download("OpenGVLab/InternVL2-2B", - allow_patterns=DOWNLOAD_PATTERN), + "OpenGVLab/InternVL2-1B", + "OpenGVLab/InternVL2-2B", # Broken due to outdated implementation of Phi-3 # See: https://huggingface.co/OpenGVLab/InternVL2-4B/discussions/3 - # snapshot_download("OpenGVLab/InternVL2-4B"), + # "OpenGVLab/InternVL2-4B", ]