Browse Source

count non-repeating vision layers

Michael Yang 1 month ago
parent
commit
8d76fa23ef
1 changed files with 1 additions and 1 deletions
  1. 1 1
      fs/ggml/ggml.go

+ 1 - 1
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()
 			}