Explorar o código

Add missing BF16 tensor type. (#7193)

Co-authored-by: Richard Lyons <frob@cloudstaff.com>
frob hai 6 meses
pai
achega
09035b71cd
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      llm/ggml.go

+ 2 - 0
llm/ggml.go

@@ -244,6 +244,8 @@ func (t Tensor) typeSize() uint64 {
 		return 8
 		return 8
 	case 29: // IQ1_M
 	case 29: // IQ1_M
 		return blockSize/8 + blockSize/16 + blockSize/32
 		return blockSize/8 + blockSize/16 + blockSize/32
+	case 30: // BF16
+		return 2
 	default:
 	default:
 		return 0
 		return 0
 	}
 	}