baseten-admin commited on
Commit
2bf0822
·
verified ·
1 Parent(s): b3842c5

Update modeling_gemma4_sequence.py

Browse files
Files changed (1) hide show
  1. modeling_gemma4_sequence.py +1 -1
modeling_gemma4_sequence.py CHANGED
@@ -198,4 +198,4 @@ class Gemma4ForSequenceClassification(Gemma4PreTrainedModel):
198
  )
199
 
200
 
201
- Gemma4ForTokenSequenceClassification.register_for_auto_class("AutoModelForSequenceClassification")
 
198
  )
199
 
200
 
201
+ Gemma4ForSequenceClassification.register_for_auto_class("AutoModelForSequenceClassification")