CERBERUS/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-10-21 19:11:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

count non-repeating vision layers

Browse source
This commit is contained in:
Michael Yang 2025-03-13 16:53:22 -07:00
parent 65b88c544f
commit 8d76fa23ef
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
fs/ggml/ggml.go
View file

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