Skip to content

Commit dbe9c0c

Browse files
aolegOleg AfoninCISC
authored
convert : support Gemma4ForCausalLM architecture (ggml-org#23682)
* convert : support Gemma4ForCausalLM architecture (ggml-org#23674) * fix indent --------- Co-authored-by: Oleg Afonin <your.email@example.com> Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>
1 parent 6fe90de commit dbe9c0c

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
@@ -74,6 +74,7 @@
7474
"Gemma3nForCausalLM": "gemma",
7575
"Gemma3nForConditionalGeneration": "gemma",
7676
"Gemma4ForConditionalGeneration": "gemma",
77+
"Gemma4ForCausalLM": "gemma",
7778
"GemmaForCausalLM": "gemma",
7879
"Glm4ForCausalLM": "glm",
7980
"Glm4MoeForCausalLM": "glm",

conversion/gemma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -614,7 +614,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
614614
yield from super().modify_tensors(data_torch, name, bid)
615615

616616

617-
@ModelBase.register("Gemma4ForConditionalGeneration")
617+
@ModelBase.register("Gemma4ForConditionalGeneration", "Gemma4ForCausalLM")
618618
class Gemma4Model(Gemma3Model):
619619
model_arch = gguf.MODEL_ARCH.GEMMA4
620620

0 commit comments

Comments
 (0)