-
Notifications
You must be signed in to change notification settings - Fork 31.4k
Closed
Labels
Feature requestRequest for a new featureRequest for a new feature
Description
Feature request
Hello, I was wondering when will you add support for Gemma3Config for sequence generation as currently only GemmaConfig and Gemma2Config are supported
Motivation
This would be extremely beneficial given that Gemma 2 2B Instruct excels as a sequence classifier. I would expect Gemma 3 4B to be even more performant.
Your contribution
I already did something to finetune Gemma 3 1B by using Gemma2ForSequenceClassification:
In the "modeling_gemma2.py", I import the Gemma 3 text model class as follows:
from ..gemma3.modeling_gemma3 import Gemma3TextModel
and then in "Gemma2ForSequenceClassification" class in the same file, I change this line:
self.model = Gemma2Model(config)
to
self.model = Gemma3TextModel(config)
kristaller486, pySilver, chuanxd, corbt, Oxi84 and 2 more
Metadata
Metadata
Assignees
Labels
Feature requestRequest for a new featureRequest for a new feature