Update convert/reader_safetensors.go

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

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
func safetensorsPad(n, s int64) int64 {
return 8 + n + s
func safetensorsPad(n, offset int64) int64 {
return 8 + n + offset
}
type safetensor struct {