count non-repeating vision layers

8d76fa23 · Michael Yang · 65b88c54 · 8d76fa23
Commit 8d76fa23 authored Mar 13, 2025 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fs/ggml/ggml.go fs/ggml/ggml.go +1 -1

No files found.
--- 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()
 			}