convert : accept "ExaoneMoeForCausalLM" arch spelling (#26660)

diff --git a/conversion/__init__.py b/conversion/__init__.py
index 06c2c50ad24571be56689aa83052b1ee17a8b299..1f781a7903af3dfde3cec33589dc808a6fb517a7 100644 (file)
--- a/conversion/__init__.py
+++ b/conversion/__init__.py
@@ -70,6 +70,7 @@ TEXT_MODEL_MAP: dict[str, str] = {
     "Exaone4ForCausalLM": "exaone",
     "ExaoneForCausalLM": "exaone",
     "ExaoneMoEForCausalLM": "exaone",
+    "ExaoneMoeForCausalLM": "exaone",
     "FalconForCausalLM": "falcon",
     "FalconH1ForCausalLM": "falcon_h1",
     "FalconMambaForCausalLM": "mamba",

diff --git a/conversion/exaone.py b/conversion/exaone.py
index bc4fb3f1b1711a29f7a86821a6346693fb605db2..1cd2244dbc3d285417dfeb643aca16f11326da79 100644 (file)
--- a/conversion/exaone.py
+++ b/conversion/exaone.py
@@ -123,7 +123,9 @@ class Exaone4Model(TextModel):
                 yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FREQS), torch.tensor(rope_factors, dtype=torch.float32))
 
 
-@ModelBase.register("ExaoneMoEForCausalLM")
+# note: transformers >= 5.1 renamed the class to "ExaoneMoeForCausalLM" (lowercase 'e'),
+#       so accept both spellings - LG AI have updated the configs of already-released models
+@ModelBase.register("ExaoneMoEForCausalLM", "ExaoneMoeForCausalLM")
 class ExaoneMoEModel(Exaone4Model):
     model_arch = gguf.MODEL_ARCH.EXAONE_MOE