Commit 47be5a1c authored by zhuwenwen's avatar zhuwenwen
Browse files

update tests

parent 7bdb7e63
......@@ -46,7 +46,8 @@ def llm():
@pytest.fixture(scope="module")
def zephyr_lora_files():
return snapshot_download(repo_id=LORA_NAME)
# return snapshot_download(repo_id=LORA_NAME)
return LORA_NAME
@pytest.mark.skip_global_cleanup
......
......@@ -4,7 +4,7 @@ import os
import pytest
import torch
import torch.nn as nn
from huggingface_hub import snapshot_download
# from huggingface_hub import snapshot_download
from transformers import AutoConfig, AutoModel, CLIPImageProcessor
from ....conftest import _ImageAssets, cleanup
......@@ -14,10 +14,8 @@ from ....utils import models_path_prefix
# dynamic_module and trust_remote_code for hf_runner
DOWNLOAD_PATTERN = ["*.json", "*.py", "*.safetensors", "*.txt", "*.model"]
models = [
snapshot_download(os.path.join(models_path_prefix, "OpenGVLab/InternViT-300M-448px"),
allow_patterns=DOWNLOAD_PATTERN),
snapshot_download(os.path.join(models_path_prefix, "OpenGVLab/InternViT-6B-448px-V1-5"),
allow_patterns=DOWNLOAD_PATTERN),
os.path.join(models_path_prefix, "OpenGVLab/InternViT-300M-448px"),
os.path.join(models_path_prefix, "OpenGVLab/InternViT-6B-448px-V1-5"),
]
......
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