diff --git a/vllm/model_executor/models/qwen2_cls.py b/vllm/model_executor/models/qwen2_cls.py index f6aec7409b2cf..cb12cce3d4780 100644 --- a/vllm/model_executor/models/qwen2_cls.py +++ b/vllm/model_executor/models/qwen2_cls.py @@ -1,9 +1,10 @@ # coding=utf-8 # Adapted from # https://huggingface.co/Qwen/Qwen2.5-Math-RM-72B/blob/main/modeling_qwen2_rm.py +# Copyright 2024 Kakao Corp(Kanana-X Team). # Copyright 2024 The Qwen team. # Copyright 2023 The vLLM team. -"""Inference-only Qwen2-RM model compatible with HuggingFace weights.""" +"""Inference-only Qwen2-Classification model compatible with HuggingFace weights.""" from typing import Iterable, List, Optional, Tuple import torch @@ -23,17 +24,6 @@ from .utils import AutoWeightsLoader -class ReLU(nn.Module): - - def __init__(self): - super().__init__() - self.activation = nn.ReLU() - - def forward(self, input): - input, _ = input - return self.activation(input) - - class Qwen2ForSequenceClassification(nn.Module): packed_modules_mapping = { "qkv_proj": [