Rename Olmo1124 to Olmo2 (#10500)

This commit is contained in:
Shane A
2024-11-25 10:36:09 -08:00
committed by GitHub
parent 10bce0450f
commit 80acb7b430
4 changed files with 29 additions and 29 deletions

View File

@@ -3040,9 +3040,9 @@ class OlmoModel(Model):
return [(self.map_tensor_name(name), data_torch)]
@Model.register("Olmo1124ForCausalLM")
class Olmo1124Model(Model):
model_arch = gguf.MODEL_ARCH.OLMO_1124
@Model.register("Olmo2ForCausalLM")
class Olmo2Model(Model):
model_arch = gguf.MODEL_ARCH.OLMO2
@Model.register("OlmoeForCausalLM")