Print quant_config

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 21, 2024 · 3025341 · 3025341
1 parent 80f5874
commit 3025341
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vllm/model_executor/models/bert.py b/vllm/model_executor/models/bert.py
@@ -390,8 +390,9 @@ def __init__(
     ) -> None:
         super().__init__()
         self.model = BertModel(config, cache_config, quant_config)
-        print("====================================================")
-        print(cache_config.__dict__)
+        print("========================      quant_config.__dict__       ============================")
+        print(quant_config.__dict__)
+        print("========================      quant_config.__dict__       ============================")
         # self.pooling_type = ModelConfig.get_pooling_type(config)
         # self._pooler = Pooler(pooling_type=self.pooling_type, normalize=True)
         self._pooler = Pooler(PoolingType.CLS, normalize=True)