From 8d76fa23ef058d98cb66ad9c3ecc5fff6d380ca1 Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Thu, 13 Mar 2025 16:53:22 -0700
Subject: [PATCH] count non-repeating vision layers

---
 fs/ggml/ggml.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fs/ggml/ggml.go b/fs/ggml/ggml.go
index 2c04559f2..dd84380b5 100644
--- a/fs/ggml/ggml.go
+++ b/fs/ggml/ggml.go
@@ -584,7 +584,7 @@ func (llm GGML) VisionGraphSize() (weights, graphSize uint64) {
 	}
 
 	for name, layer := range llm.Tensors().GroupLayers() {
-		if strings.HasPrefix(name, "v.") {
+		if name == "v" || strings.HasPrefix(name, "v.") {
 			for _, tensor := range layer {
 				weights += tensor.Size()
 			}