biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix/remove some broken model executor tests (#21224)

Browse Source 
Signed-off-by: Rabi Mishra <ramishra@redhat.com>
This commit is contained in:
Rabi Mishra
2025-07-19 17:45:07 +05:30
committed by GitHub
parent e3a0e43d7f
commit c81259d33a
2 changed files with 3 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/model_executor/test_model_load_with_params.py
View File

@@ -49,7 +49,7 @@ def test_model_loading_with_params(vllm_runner):
def check_model(model):
assert isinstance(model, BertEmbeddingModel)
assert isinstance(model._pooler, CLSPool)
assert isinstance(model.pooler.pooling, CLSPool)
vllm_model.apply_model(check_model)
@@ -87,7 +87,7 @@ def test_roberta_model_loading_with_params(vllm_runner):
def check_model(model):
assert isinstance(model, RobertaEmbeddingModel)
assert isinstance(model._pooler, MeanPool)
assert isinstance(model.pooler.pooling, MeanPool)
vllm_model.apply_model(check_model)
@@ -114,7 +114,7 @@ def test_facebook_roberta_model_loading_with_params(vllm_runner):
def check_model(model):
assert isinstance(model, RobertaEmbeddingModel)
assert not hasattr(model, "lm_head")
assert isinstance(model._pooler, CLSPool)
assert isinstance(model.pooler.pooling, CLSPool)
vllm_model.apply_model(check_model)