diff --git a/gguf-py/gguf/constants.py b/gguf-py/gguf/constants.py index 46cc6a143..d99616d30 100644 --- a/gguf-py/gguf/constants.py +++ b/gguf-py/gguf/constants.py @@ -95,7 +95,6 @@ class MODEL_ARCH(IntEnum): BLOOM = auto() STABLELM = auto() QWEN = auto() - PHI2 = auto() PHI = auto() @@ -142,7 +141,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = { MODEL_ARCH.BLOOM: "bloom", MODEL_ARCH.STABLELM: "stablelm", MODEL_ARCH.QWEN: "qwen", - MODEL_ARCH.PHI2: "phi2", + MODEL_ARCH.PHI: "phi", } TENSOR_NAMES: dict[MODEL_TENSOR, str] = { @@ -353,7 +352,7 @@ MODEL_TENSORS: dict[MODEL_ARCH, list[MODEL_TENSOR]] = { MODEL_ARCH.GPT2: [ # TODO ], - MODEL_ARCH.PHI2: [ + MODEL_ARCH.PHI: [ MODEL_TENSOR.TOKEN_EMBD, MODEL_TENSOR.OUTPUT_NORM, MODEL_TENSOR.OUTPUT, @@ -364,9 +363,6 @@ MODEL_TENSORS: dict[MODEL_ARCH, list[MODEL_TENSOR]] = { MODEL_TENSOR.FFN_DOWN, MODEL_TENSOR.FFN_UP, ] - MODEL_ARCH.PHI: [ - # TODO - ], # TODO }