[Bugfix] Fix QwenModel argument (#10262)

Signed-off-by: Jie Fu <jiefu@tencent.com>
This commit is contained in:
Jie Fu (傅杰) 2024-11-13 00:36:51 +08:00 committed by GitHub
parent a838ba7254
commit 176fcb1c71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1068,7 +1068,7 @@ class QWenLMHeadModel(QWenBaseModel, SupportsLoRA):
config = vllm_config.model_config.hf_config config = vllm_config.model_config.hf_config
# Initialize VL # Initialize VL
if hasattr(config, "visual"): if hasattr(config, "visual"):
return QWenVL(vllm_config) return QWenVL(vllm_config=vllm_config)
# Initialize LLM # Initialize LLM
else: else:
return QWenLLM(vllm_config) return QWenLLM(vllm_config=vllm_config)