File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
tests/models/multimodal/pooling Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -33,8 +33,10 @@ def _run_test(
33
33
skip_tokenizer_init = True ) as vllm_model :
34
34
vllm_model .encode (prompt )
35
35
36
+
36
37
MODELS = ["christian-pinto/Prithvi-EO-2.0-300M-TL-VLLM" ]
37
38
39
+
38
40
@pytest .mark .parametrize ("model" , MODELS )
39
41
def test_models_image (
40
42
hf_runner ,
Original file line number Diff line number Diff line change @@ -612,8 +612,6 @@ def __post_init__(self) -> None:
612
612
self .served_model_name = get_served_model_name (self .model ,
613
613
self .served_model_name )
614
614
self .multimodal_config = self ._init_multimodal_config ()
615
- self .is_pooling_model = self .registry .is_pooling_model (
616
- self .architectures )
617
615
self .model_supports_multimodal_raw_input = (
618
616
self ._init_model_supports_multimodal_raw_input ())
619
617
if not self .skip_tokenizer_init :
@@ -1424,6 +1422,10 @@ def uses_mrope(self) -> bool:
1424
1422
@property
1425
1423
def is_multimodal_model (self ) -> bool :
1426
1424
return self .multimodal_config is not None
1425
+
1426
+ @property
1427
+ def is_pooling_model (self ) -> bool :
1428
+ return self .registry .is_pooling_model (self .architectures )
1427
1429
1428
1430
@property
1429
1431
def is_cross_encoder (self ) -> bool :
You can’t perform that action at this time.
0 commit comments