vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix formatting in gguf.py

Browse source
This commit is contained in:
Galunid 2023-10-22 11:08:19 +02:00
parent 76b4495cec
commit e167ebcb22
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
gguf-py/gguf/gguf.py
View file

@ -89,7 +89,7 @@ class MODEL_ARCH(IntEnum):
REFACT : int = auto()
BERT : int = auto()
BLOOM : int = auto()
STABLELM : int = auto()
STABLELM : int = auto()
class MODEL_TENSOR(IntEnum):
@ -129,7 +129,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = {
MODEL_ARCH.REFACT: "refact",
MODEL_ARCH.BERT: "bert",
MODEL_ARCH.BLOOM: "bloom",
MODEL_ARCH.STABLELM: "stablelm",
MODEL_ARCH.STABLELM: "stablelm",
}
TENSOR_NAMES: dict[MODEL_TENSOR, str] = {