diff --git a/conversion/__init__.py b/conversion/__init__.py index 8a549ce28f..308bdaff44 100644 --- a/conversion/__init__.py +++ b/conversion/__init__.py @@ -79,6 +79,7 @@ TEXT_MODEL_MAP: dict[str, str] = { "Gemma4ForConditionalGeneration": "gemma", "Gemma4ForCausalLM": "gemma", "Gemma4UnifiedForConditionalGeneration": "gemma", + "Gemma4UnifiedAssistantForCausalLM": "gemma", "GemmaForCausalLM": "gemma", "Glm4ForCausalLM": "glm", "Glm4MoeForCausalLM": "glm", diff --git a/conversion/gemma.py b/conversion/gemma.py index bb68aedd59..0e3f953a01 100644 --- a/conversion/gemma.py +++ b/conversion/gemma.py @@ -785,7 +785,7 @@ class Gemma4UnifiedModel(Gemma4Model): self.gguf_writer.add_suppress_tokens(suppress_tokens) -@ModelBase.register("Gemma4AssistantForCausalLM") +@ModelBase.register("Gemma4AssistantForCausalLM", "Gemma4UnifiedAssistantForCausalLM") class Gemma4AssistantModel(Gemma4Model): model_arch = gguf.MODEL_ARCH.GEMMA4_ASSISTANT