fixes param

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 21, 2024 · c3820f8 · c3820f8
1 parent 7f4a1ce
commit c3820f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)