Skip to content

Commit eb783f0

Browse files
Some style changes
- Improved formatting around - made is_pooling_model a @Property in ModelConfig Signed-off-by: Christian Pinto <christian.pinto@ibm.com>
1 parent 9e98f1c commit eb783f0

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

tests/models/multimodal/pooling/test_prithvi_mae.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,10 @@ def _run_test(
3333
skip_tokenizer_init=True) as vllm_model:
3434
vllm_model.encode(prompt)
3535

36+
3637
MODELS = ["christian-pinto/Prithvi-EO-2.0-300M-TL-VLLM"]
3738

39+
3840
@pytest.mark.parametrize("model", MODELS)
3941
def test_models_image(
4042
hf_runner,

vllm/config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -612,8 +612,6 @@ def __post_init__(self) -> None:
612612
self.served_model_name = get_served_model_name(self.model,
613613
self.served_model_name)
614614
self.multimodal_config = self._init_multimodal_config()
615-
self.is_pooling_model = self.registry.is_pooling_model(
616-
self.architectures)
617615
self.model_supports_multimodal_raw_input = (
618616
self._init_model_supports_multimodal_raw_input())
619617
if not self.skip_tokenizer_init:
@@ -1424,6 +1422,10 @@ def uses_mrope(self) -> bool:
14241422
@property
14251423
def is_multimodal_model(self) -> bool:
14261424
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)
14271429

14281430
@property
14291431
def is_cross_encoder(self) -> bool:

0 commit comments

Comments
 (0)