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

Update convert/reader_safetensors.go

Browse source 
Co-authored-by: Jeffrey Morgan <jmorganca@gmail.com>
This commit is contained in:
Michael Yang 2024-07-31 10:58:22 -07:00
parent df993fa37b
commit 781fc2d576
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
convert/reader_safetensors.go
View file

@ -67,8 +67,8 @@ func parseSafetensors(ps ...string) ([]Tensor, error) {
} }
// safetensorsPad returns the padded size of the safetensors file given a length n and offset s // safetensorsPad returns the padded size of the safetensors file given a length n and offset s
func safetensorsPad(n, s int64) int64 { func safetensorsPad(n, offset int64) int64 {
return 8 + n + s return 8 + n + offset
} }
type safetensor struct { type safetensor struct {