[Model] Support Qwen2ForSequenceClassification (#4609)

Co-authored-by: ximing.wxm <ximing.wxm@antgroup.com>
2025-03-25 10:13:44 +08:00
parent 4c584fc632
commit 22c3702e1e
4 changed files with 87 additions and 2 deletions
--- a/python/sglang/srt/configs/model_config.py
+++ b/python/sglang/srt/configs/model_config.py
@@ -453,6 +453,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 "Qwen2ForSequenceClassification" in model_architectures
    ):
        return False
    else: