From 80a84bbe09ab006cf877a6ad6303cc0503125da7 Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Tue, 22 Oct 2024 11:16:31 -0300 Subject: [PATCH] Linting fix Signed-off-by: Flavia Beo --- vllm/model_executor/layers/pooler.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vllm/model_executor/layers/pooler.py b/vllm/model_executor/layers/pooler.py index e94501cdb8b0d..a447d630e8f9f 100644 --- a/vllm/model_executor/layers/pooler.py +++ b/vllm/model_executor/layers/pooler.py @@ -16,6 +16,7 @@ class PoolingType(IntEnum): CLS = 2 MEAN = 3 + @dataclass class PoolingConfig: """A class that configures the pooling operation which