Skip to content

GitLab

"...text-generation-inference.git" did not exist on "5a65066922ce28dbc202dc03bb2410da14b980d2"
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) { ...@@ -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 {
......
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