Skip to content

Commit

Permalink
remove print
Browse files Browse the repository at this point in the history
Signed-off-by: Flavia Beo <[email protected]>
  • Loading branch information
flaviabeo committed Oct 21, 2024
1 parent 5f08d87 commit 66171c4
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions vllm/model_executor/models/bert.py
Original file line number Diff line number Diff line change
Expand Up @@ -390,10 +390,6 @@ def __init__(
) -> None:
super().__init__()
self.model = BertModel(config, cache_config, quant_config)
print("======================== quant_config.__dict__ ============================")
print(quant_config.__dict__)
print(quant_config.get_config_filenames())
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)
Expand Down

0 comments on commit 66171c4

Please sign in to comment.