[Bugfix] Fix QwenModel argument (#10262)

Signed-off-by: Jie Fu <jiefu@tencent.com>
2024-11-13 00:36:51 +08:00 · 2024-11-13 00:36:51 +08:00 · 176fcb1c71
commit 176fcb1c71
parent a838ba7254
1 changed files with 2 additions and 2 deletions
--- a/vllm/model_executor/models/qwen.py
+++ b/vllm/model_executor/models/qwen.py
@ -1068,7 +1068,7 @@ class QWenLMHeadModel(QWenBaseModel, SupportsLoRA):
        config = vllm_config.model_config.hf_config
        # Initialize VL
        if hasattr(config, "visual"):
-            return QWenVL(vllm_config)
+            return QWenVL(vllm_config=vllm_config)
        # Initialize LLM
        else:
-            return QWenLLM(vllm_config)
+            return QWenLLM(vllm_config=vllm_config)