mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-08-07 23:45:30 +00:00
convert : accept "ExaoneMoeForCausalLM" arch spelling (#26660)
This commit is contained in:
parent
a1f96d4fc2
commit
e700bfb37f
2 changed files with 4 additions and 1 deletions
|
|
@ -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",
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue