Explorar el Código

count non-repeating vision layers

Michael Yang hace 1 mes
padre
commit
8d76fa23ef
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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()
 			}