baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add missing BF16 tensor type. (#7193)

Browse source 
Co-authored-by: Richard Lyons <frob@cloudstaff.com>
This commit is contained in:
frob 2024-10-15 02:06:35 +02:00 committed by GitHub
parent f3c8b898cd
commit 09035b71cd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
llm/ggml.go
View file

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