from transformers import Gemma3Config class CustomGemmaConfig(Gemma3Config): model_type = "custom_gemma"