Skip to content

Commit 4eaa3ce

Browse files
committed
add unified assistant
1 parent dd97604 commit 4eaa3ce

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

conversion/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@
7979
"Gemma4ForConditionalGeneration": "gemma",
8080
"Gemma4ForCausalLM": "gemma",
8181
"Gemma4UnifiedForConditionalGeneration": "gemma",
82+
"Gemma4UnifiedAssistantForCausalLM": "gemma",
8283
"GemmaForCausalLM": "gemma",
8384
"Glm4ForCausalLM": "glm",
8485
"Glm4MoeForCausalLM": "glm",

conversion/gemma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -785,7 +785,7 @@ def set_gguf_parameters(self):
785785
self.gguf_writer.add_suppress_tokens(suppress_tokens)
786786

787787

788-
@ModelBase.register("Gemma4AssistantForCausalLM")
788+
@ModelBase.register("Gemma4AssistantForCausalLM", "Gemma4UnifiedAssistantForCausalLM")
789789
class Gemma4AssistantModel(Gemma4Model):
790790
model_arch = gguf.MODEL_ARCH.GEMMA4_ASSISTANT
791791

0 commit comments

Comments
 (0)