Skip to content

GitLab

Commit 781fc2d5 authored by Michael Yang's avatar Michael Yang
Browse files

Update convert/reader_safetensors.go 


Co-authored-by: default avatarJeffrey Morgan <jmorganca@gmail.com>
parent df993fa3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
convert/reader_safetensors.go
View file @ 781fc2d5
......@@ -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 {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment