[Model] Support Qwen3ForSequenceClassification for Qwen3-Embed Model (#7957)

This commit is contained in:
Zhihao Liu
2025-08-14 02:14:54 +08:00
committed by GitHub
parent 7b56e494be
commit 65736dc524
3 changed files with 166 additions and 54 deletions

View File

@@ -642,6 +642,7 @@ def is_generation_model(model_architectures: List[str], is_embedding: bool = Fal
or "InternLM2ForRewardModel" in model_architectures
or "Qwen2ForRewardModel" in model_architectures
or "Qwen2ForSequenceClassification" in model_architectures
or "Qwen3ForSequenceClassification" in model_architectures
or "CLIPModel" in model_architectures
or "BertModel" in model_architectures
or "Contriever" in model_architectures