convert : support Gemma4ForCausalLM architecture (#23682)

* convert : support Gemma4ForCausalLM architecture (#23674)

* fix indent

---------

Co-authored-by: Oleg Afonin <redacted>
Co-authored-by: Sigbjørn Skjæret <redacted>

diff --git a/conversion/__init__.py b/conversion/__init__.py
index 73e2b3639e0e8df06e282f332e87116e72a4f5a8..2a87bd75b441e9c09c9dd9d0659224f752698905 100644 (file)
--- a/conversion/__init__.py
+++ b/conversion/__init__.py
@@ -74,6 +74,7 @@ TEXT_MODEL_MAP: dict[str, str] = {
     "Gemma3nForCausalLM": "gemma",
     "Gemma3nForConditionalGeneration": "gemma",
     "Gemma4ForConditionalGeneration": "gemma",
+    "Gemma4ForCausalLM": "gemma",
     "GemmaForCausalLM": "gemma",
     "Glm4ForCausalLM": "glm",
     "Glm4MoeForCausalLM": "glm",

diff --git a/conversion/gemma.py b/conversion/gemma.py
index a6e14fbcb9828039321abaf6fb99fd7ca7778d5a..1b427a30cd5cd4d046718a3ddc8f6ae3b0575fc6 100644 (file)
--- a/conversion/gemma.py
+++ b/conversion/gemma.py
@@ -614,7 +614,7 @@ class Gemma3NModel(Gemma3Model):
         yield from super().modify_tensors(data_torch, name, bid)
 
 
-@ModelBase.register("Gemma4ForConditionalGeneration")
+@ModelBase.register("Gemma4ForConditionalGeneration", "Gemma4ForCausalLM")
 class Gemma4Model(Gemma3Model):
     model_arch = gguf.MODEL_ARCH.GEMMA4