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

@@ -642,8 +642,6 @@ def __post_init__(self) -> None:
         self.original_max_model_len = self.max_model_len
         self.max_model_len = self.get_and_verify_max_len(self.max_model_len)
         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:
@@ -1516,6 +1514,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: