Skip to content

Commit e0ae311

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 0dba4cd commit e0ae311

File tree

3 files changed

+7
-2
lines changed

3 files changed

+7
-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:

vllm/v1/core/kv_cache_manager.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def new_empty(self) -> "KVCacheBlocks":
6767

6868
class KVCacheManagerInterface(ABC):
6969

70+
@property
7071
@abstractmethod
7172
def usage(self) -> float:
7273
raise NotImplementedError

0 commit comments

Comments
 (0)