From 94b74079ed3ff1f2bcc01eea4e471d329aea0a64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=A7=BB=E5=B1=B1=E6=90=AC=E7=A0=96=E6=B4=BE?= Date: Wed, 7 May 2025 11:51:43 +0800 Subject: [PATCH] Update registry.py update registry.py to support Qwen3 Embedding Model --- vllm/model_executor/models/registry.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vllm/model_executor/models/registry.py b/vllm/model_executor/models/registry.py index 19153efd8e17..fc1de00906b3 100644 --- a/vllm/model_executor/models/registry.py +++ b/vllm/model_executor/models/registry.py @@ -152,6 +152,7 @@ "Qwen2VLForConditionalGeneration": ("qwen2_vl", "Qwen2VLForConditionalGeneration"), # noqa: E501 # [Auto-converted (see adapters.py)] "Qwen2ForSequenceClassification": ("qwen2", "Qwen2ForCausalLM"), + "Qwen3ForSequenceClassification": ("qwen3", "Qwen3ForCausalLM"), # Technically PrithviGeoSpatialMAE is a model that works on images, both in # input and output. I am adding it here because it piggy-backs on embedding # models for the time being.