Some style changes

- Improved formatting around
- made is_pooling_model a @Property in ModelConfig

Signed-off-by: Christian Pinto <christian.pinto@ibm.com>

Author: christian-pinto

tests/models/multimodal/pooling/test_prithvi_mae.py

@@ -33,8 +33,10 @@ def _run_test(
                      skip_tokenizer_init=True) as vllm_model:
         vllm_model.encode(prompt)
 
+
 MODELS = ["christian-pinto/Prithvi-EO-2.0-300M-TL-VLLM"]
 
+
 @pytest.mark.parametrize("model", MODELS)
 def test_models_image(
     hf_runner,

vllm/config.py

@@ -614,8 +614,6 @@ def __post_init__(self) -> None:
         self.served_model_name = get_served_model_name(self.model,
                                                        self.served_model_name)
         self.multimodal_config = self._init_multimodal_config()
-        self.is_pooling_model = self.registry.is_pooling_model(
-            self.architectures)
         self.model_supports_multimodal_raw_input = (
             self._init_model_supports_multimodal_raw_input())
         if not self.skip_tokenizer_init:
@@ -1404,6 +1402,10 @@ def uses_mrope(self) -> bool:
     @property
     def is_multimodal_model(self) -> bool:
         return self.multimodal_config is not None
+    
+    @property
+    def is_pooling_model(self) -> bool: 
+        return self.registry.is_pooling_model(self.architectures)
 
     @property
     def is_cross_encoder(self) -> bool:

vllm/v1/core/kv_cache_manager.py

@@ -67,6 +67,7 @@ def new_empty(self) -> "KVCacheBlocks":
 
 class KVCacheManagerInterface(ABC):
 
+    @property
     @abstractmethod
     def usage(self) -> float:
         raise NotImplementedError