param default value

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 21, 2024 · 66152c2 · 66152c2
1 parent d371268
commit 66152c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/model_executor/models/bert.py b/vllm/model_executor/models/bert.py
@@ -387,7 +387,7 @@ def __init__(
         config: BertConfig,
         cache_config: Optional[CacheConfig] = None,
         quant_config: Optional[QuantizationConfig] = None,
-        pooling_type
+        pooling_type: Optional = None
     ) -> None:
         super().__init__()
         self.model = BertModel(config, cache_config, quant_config)