diff --git a/conversion/__init__.py b/conversion/__init__.py index 06c2c50ad245..1f781a7903af 100644 --- a/conversion/__init__.py +++ b/conversion/__init__.py @@ -70,6 +70,7 @@ "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 bc4fb3f1b171..1cd2244dbc3d 100644 --- a/conversion/exaone.py +++ b/conversion/exaone.py @@ -123,7 +123,9 @@ def generate_extra_tensors(self) -> Iterable[tuple[str, Tensor]]: 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