From c3820f8b494319da729ff587ce52f864c998f59d Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Mon, 21 Oct 2024 11:35:25 -0300 Subject: [PATCH] fixes param Signed-off-by: Flavia Beo --- vllm/model_executor/model_loader/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/model_loader/loader.py b/vllm/model_executor/model_loader/loader.py index 764c91f4aa352..ec68c1b5345a7 100644 --- a/vllm/model_executor/model_loader/loader.py +++ b/vllm/model_executor/model_loader/loader.py @@ -163,7 +163,7 @@ def build_model(model_class: Type[nn.Module], hf_config: PretrainedConfig, return model_class(config=hf_config, cache_config=cache_config, quant_config=quant_config, - pooling_type, + pooling_type=pooling_type, **extra_kwargs)