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

update tests

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