Skip to content

Commit 0298ff8

Browse files
committed
+ GemmaForSequenceClassification
Signed-off-by: wang.yuqi <noooop@126.com>
1 parent 6e244ae commit 0298ff8

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

vllm/model_executor/models/gemma.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
from vllm.model_executor.sampling_metadata import SamplingMetadata
4444
from vllm.sequence import IntermediateTensors
4545

46+
from .adapters import as_seq_cls_model
4647
from .interfaces import SupportsLoRA, SupportsPP
4748
from .utils import (AutoWeightsLoader, is_pp_missing_parameter,
4849
make_empty_intermediate_tensors_factory, make_layers,
@@ -425,3 +426,7 @@ def load_weights(self, weights: Iterable[tuple[str,
425426
if self.config.tie_word_embeddings else None),
426427
)
427428
return loader.load_weights(weights)
429+
430+
431+
class GemmaForSequenceClassification(as_seq_cls_model(GemmaForCausalLM)):
432+
pass

vllm/model_executor/models/registry.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,8 @@
174174
"RobertaForSequenceClassification"),
175175
"ModernBertForSequenceClassification": ("modernbert",
176176
"ModernBertForSequenceClassification"),
177+
# [Auto-converted (see adapters.py)]
178+
"GemmaForSequenceClassification": ("gemma", "GemmaForSequenceClassification"), # noqa: E501
177179
"Qwen3ForSequenceClassification": ("qwen3", "Qwen3ForSequenceClassification"), # noqa: E501
178180
}
179181

0 commit comments

Comments
 (0)