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