From fd5e2268a593912dac9c390bdaefc6b2a881f3d0 Mon Sep 17 00:00:00 2001
From: Cebtenzzre <cebtenzzre@gmail.com>
Date: Sat, 30 Sep 2023 19:47:06 -0400
Subject: [PATCH] gguf : fix typos

---
 gguf-py/gguf/gguf.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/gguf-py/gguf/gguf.py b/gguf-py/gguf/gguf.py
index a90d003c0..a291f6796 100644
--- a/gguf-py/gguf/gguf.py
+++ b/gguf-py/gguf/gguf.py
@@ -120,6 +120,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = {
 
 TENSOR_NAMES: dict[MODEL_TENSOR, str] = {
     MODEL_TENSOR.TOKEN_EMBD:    "token_embd",
+    MODEL_TENSOR.POS_EMBD:      "position_embd",
     MODEL_TENSOR.OUTPUT_NORM:   "output_norm",
     MODEL_TENSOR.OUTPUT:        "output",
     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]] = {
-    {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