Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
47be5a1c
Commit
47be5a1c
authored
Dec 03, 2024
by
zhuwenwen
Browse files
update tests
parent
7bdb7e63
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
6 deletions
+5
-6
tests/entrypoints/llm/test_generate_multiple_loras.py
tests/entrypoints/llm/test_generate_multiple_loras.py
+2
-1
tests/models/decoder_only/vision_language/test_intern_vit.py
tests/models/decoder_only/vision_language/test_intern_vit.py
+3
-5
No files found.
tests/entrypoints/llm/test_generate_multiple_loras.py
View file @
47be5a1c
...
...
@@ -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
...
...
tests/models/decoder_only/vision_language/test_intern_vit.py
View file @
47be5a1c
...
...
@@ -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"
),
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment