diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index ddb7ec8f..59803bd6 100644 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -3474,7 +3474,7 @@ class Gemma4Model(Gemma4BaseModel): return [(self.map_tensor_name(name), data_torch)] -@Model.register("Gemma4AssistantForCausalLM") +@Model.register("Gemma4AssistantForCausalLM", "Gemma4UnifiedAssistantForCausalLM") class Gemma4AssistantModel(Gemma4BaseModel): model_arch = gguf.MODEL_ARCH.GEMMA4_MTP