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

gguf : fix typos

Browse source
This commit is contained in:
Cebtenzzre 2023-09-30 19:47:06 -04:00
parent 84f7cea2f9
commit fd5e2268a5
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
gguf-py/gguf/gguf.py
View file

@ -120,6 +120,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = {
TENSOR_NAMES: dict[MODEL_TENSOR, str] = { TENSOR_NAMES: dict[MODEL_TENSOR, str] = {
MODEL_TENSOR.TOKEN_EMBD: "token_embd", MODEL_TENSOR.TOKEN_EMBD: "token_embd",
MODEL_TENSOR.POS_EMBD: "position_embd",
MODEL_TENSOR.OUTPUT_NORM: "output_norm", MODEL_TENSOR.OUTPUT_NORM: "output_norm",
MODEL_TENSOR.OUTPUT: "output", MODEL_TENSOR.OUTPUT: "output",
MODEL_TENSOR.ROPE_FREQS: "rope_freqs", MODEL_TENSOR.ROPE_FREQS: "rope_freqs",
@ -212,7 +213,7 @@ MODEL_TENSORS: dict[MODEL_ARCH, list[MODEL_TENSOR]] = {
} }
MODEL_TENSOR_NAMES: dict[MODEL_ARCH, dict[MODEL_TENSOR, str]] = { MODEL_TENSOR_NAMES: dict[MODEL_ARCH, dict[MODEL_TENSOR, str]] = {
{t: TENSOR_NAMES[t] for t in ts} for m, ts in MODEL_TENSORS.items() m: {t: TENSOR_NAMES[t] for t in ts} for m, ts in MODEL_TENSORS.items()
} }
# tensors that will not be serialized # tensors that will not be serialized