Update modeling_gemma4_sequence.py

modeling_gemma4_sequence.py

@@ -15,7 +15,7 @@ from transformers.models.gemma4.configuration_gemma4 import Gemma4Config
 from transformers.models.gemma4.modeling_gemma4 import Gemma4Model, Gemma4PreTrainedModel
 
 
-class Gemma4ForTokenSequenceClassification(Gemma4PreTrainedModel):
+class Gemma4ForSequenceClassification(Gemma4PreTrainedModel):
     """Pool the last text position and score it with selected Gemma 4 token rows."""
 
     config_class = Gemma4Config
@@ -49,7 +49,7 @@ class Gemma4ForTokenSequenceClassification(Gemma4PreTrainedModel):
         model_lm: nn.Module,
         selected_token_ids: Sequence[int],
         labels: Sequence[str],
-    ) -> "Gemma4ForTokenSequenceClassification":
+    ) -> "Gemma4ForSequenceClassification":
         token_ids = torch.tensor(selected_token_ids, device=model_lm.lm_head.weight.device)
         classifier_weight = model_lm.lm_head.weight.index_select(0, token_ids).detach().clone()
         cls.configure_classification_config(model_lm.config, selected_token_ids, labels)