Update convert/reader_safetensors.go
Co-authored-by: Jeffrey Morgan <jmorganca@gmail.com>
This commit is contained in:
parent
df993fa37b
commit
781fc2d576
1 changed files with 2 additions and 2 deletions
|
@ -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 {
|
||||
|
|
Loading…
Reference in a new issue