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
2e33fe41
Unverified
Commit
2e33fe41
authored
Dec 11, 2024
by
Cyrus Leung
Committed by
GitHub
Dec 11, 2024
Browse files
[CI/Build] Check transformers v4.47 (#10991)
Signed-off-by:
DarkLight1337
<
tlleungac@connect.ust.hk
>
parent
e39400a4
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
12 deletions
+3
-12
requirements-test.txt
requirements-test.txt
+2
-2
tests/models/decoder_only/vision_language/mm_processor_kwargs/test_idefics3.py
...only/vision_language/mm_processor_kwargs/test_idefics3.py
+0
-9
tests/models/embedding/vision_language/test_llava_next.py
tests/models/embedding/vision_language/test_llava_next.py
+1
-1
No files found.
requirements-test.txt
View file @
2e33fe41
...
...
@@ -506,7 +506,7 @@ tiktoken==0.7.0
# mistral-common
timm==1.0.11
# via -r requirements-test.in
tokenizers==0.2
0.3
tokenizers==0.2
1.0
# via transformers
torch==2.5.1
# via
...
...
@@ -534,7 +534,7 @@ tqdm==4.66.6
# transformers
tqdm-multiprocess==0.0.11
# via lm-eval
transformers==4.4
6.3
transformers==4.4
7.0
# via
# lm-eval
# peft
...
...
tests/models/decoder_only/vision_language/mm_processor_kwargs/test_idefics3.py
View file @
2e33fe41
...
...
@@ -3,7 +3,6 @@ from typing import Optional
import
pytest
import
torch
import
transformers
from
transformers
import
AutoImageProcessor
,
AutoTokenizer
from
vllm.inputs
import
InputContext
,
token_inputs
...
...
@@ -36,8 +35,6 @@ def get_max_idefics3_image_tokens():
return
get_max_idefics3_image_tokens
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
<
"4.46.0"
,
reason
=
"Model introduced in HF >= 4.46.0"
)
@
pytest
.
mark
.
parametrize
(
"model"
,
models
)
@
pytest
.
mark
.
parametrize
(
"longest_edge"
,
[
None
,
168
,
336
,
400
,
2
*
336
])
def
test_input_mapper_override
(
model
:
str
,
image_assets
:
_ImageAssets
,
...
...
@@ -77,8 +74,6 @@ def test_input_mapper_override(model: str, image_assets: _ImageAssets,
assert
torch
.
all
(
hf_result
[
"pixel_values"
]
==
vllm_result
[
"pixel_values"
])
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
<
"4.46.0"
,
reason
=
"Model introduced in HF >= 4.46.0"
)
@
pytest
.
mark
.
parametrize
(
"model"
,
models
)
@
pytest
.
mark
.
parametrize
(
"longest_edge, expected_max_tokens"
,
[
(
None
,
2873
),
...
...
@@ -107,8 +102,6 @@ def test_max_tokens_override(get_max_idefics3_image_tokens, model: str,
assert
expected_max_tokens
==
actual_max_tokens
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
<
"4.46.0"
,
reason
=
"Model introduced in HF >= 4.46.0"
)
@
pytest
.
mark
.
parametrize
(
"model"
,
models
)
@
pytest
.
mark
.
parametrize
(
"longest_edge, toks_per_img, num_imgs"
,
[
(
168
,
169
,
1
),
...
...
@@ -143,8 +136,6 @@ def test_dummy_data_override(dummy_data_for_idefics3, model: str,
assert
img_tok_count
==
toks_per_img
*
num_imgs
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
<
"4.46.0"
,
reason
=
"Model introduced in HF >= 4.46.0"
)
@
pytest
.
mark
.
parametrize
(
"model"
,
models
)
@
pytest
.
mark
.
parametrize
(
"longest_edge,expected_toks_per_img,num_imgs"
,
[
(
336
,
169
*
(
1
**
2
+
1
),
1
),
...
...
tests/models/embedding/vision_language/test_llava_next.py
View file @
2e33fe41
...
...
@@ -86,7 +86,7 @@ def _run_test(
)
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
.
startswith
(
"4.46"
)
,
@
pytest
.
mark
.
skipif
(
transformers
.
__version__
>=
"4.46"
,
reason
=
"Model broken with changes in transformers 4.46"
)
@
pytest
.
mark
.
core_model
@
pytest
.
mark
.
parametrize
(
"model"
,
MODELS
)
...
...
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