Unverified Commit 9c62db07 authored by Jie Fu (傅杰)'s avatar Jie Fu (傅杰) Committed by GitHub
Browse files

[Model] Support Qwen-VL and Qwen-VL-Chat models with text-only inputs (#5710)


Co-authored-by: default avatarRoger Wang <ywang@roblox.com>
parent cf90ae01
...@@ -28,6 +28,7 @@ from vllm.model_executor.layers.vocab_parallel_embedding import ( ...@@ -28,6 +28,7 @@ from vllm.model_executor.layers.vocab_parallel_embedding import (
from vllm.model_executor.model_loader.weight_utils import default_weight_loader from vllm.model_executor.model_loader.weight_utils import default_weight_loader
from vllm.model_executor.sampling_metadata import SamplingMetadata from vllm.model_executor.sampling_metadata import SamplingMetadata
from vllm.sequence import SamplerOutput from vllm.sequence import SamplerOutput
from vllm.utils import print_warning_once
class QWenMLP(nn.Module): class QWenMLP(nn.Module):
...@@ -288,6 +289,15 @@ class QWenLMHeadModel(nn.Module): ...@@ -288,6 +289,15 @@ class QWenLMHeadModel(nn.Module):
# Skip loading extra bias for GPTQ models. # Skip loading extra bias for GPTQ models.
if name.endswith(".bias") and name not in params_dict: if name.endswith(".bias") and name not in params_dict:
continue continue
# Skip loading visual weights to support Qwen-VL models
# in cases with text-only inputs
# TODO: add support for Qwen-VL
if (name not in params_dict
and name.startswith("transformer.visual.")):
print_warning_once(
"Only text inputs are allowed. Images won't be handled "
"until Qwen-VL models are fully supported.")
continue
param = params_dict[name] param = params_dict[name]
weight_loader = getattr(param, "weight_loader", weight_loader = getattr(param, "weight_loader",
default_weight_loader) default_weight_loader)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment