[Model] Add Qwen3ForRewardModel and fix Qwen3ForSequenceClassification (#17992)

Co-authored-by: yes-its-shivam <yes-its-shivam@users.noreply.github.com>
This commit is contained in:
Shivam jindal
2026-02-16 17:14:41 +05:30
committed by GitHub
parent de833f9e8e
commit 4f0409f8aa
4 changed files with 140 additions and 30 deletions

View File

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