From 5a5dca13b2e94e693d445840cb196928878a4af9 Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Wed, 29 Nov 2023 10:54:23 -0800
Subject: [PATCH] comments

---
 llm/gguf.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/llm/gguf.go b/llm/gguf.go
index f68b87b27f3..29270e51de3 100644
--- a/llm/gguf.go
+++ b/llm/gguf.go
@@ -108,8 +108,7 @@ func (llm *ggufModel) NumKV() uint64 {
 }
 
 func (llm *ggufModel) ModelFamily() string {
-	t, ok := llm.kv["general.architecture"].(string)
-	if ok {
+	if t, ok := llm.kv["general.architecture"].(string); ok {
 		return t
 	}
 
@@ -146,8 +145,7 @@ func (llm *ggufModel) ModelType() string {
 }
 
 func (llm *ggufModel) FileType() string {
-	t, ok := llm.kv["general.file_type"].(uint32)
-	if ok {
+	if t, ok := llm.kv["general.file_type"].(uint32); ok {
 		return fileType(t)
 	}
 
@@ -216,6 +214,7 @@ func (llm *ggufModel) Decode(rso *readSeekOffset) error {
 			return err
 		}
 
+		// dims is the number of dimensions in the tensor
 		dims := llm.readU32(rso)
 
 		shape := [4]uint64{1, 1, 1, 1}