Procházet zdrojové kódy

count non-repeating vision layers

Michael Yang před 1 měsícem
rodič
revize
8d76fa23ef
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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()
 			}