mirror
/
ollama
mirror of https://github.com/ollama/ollama
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

convert: check file size for safetensors to warn for improper conversion

This commit is contained in:
ParthSareen 2025-12-10 17:58:16 -08:00
parent 1c4e85b4df
commit 5d3eeb43c0
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
convert/reader_safetensors.go
View File

@ -37,6 +37,10 @@ func parseSafetensors(fsys fs.FS, replacer *strings.Replacer, ps ...string) ([]T
return nil, err return nil, err
} }
if n <= 0 || n > 100<<20 {
return nil, fmt.Errorf("invalid safetensors file %q (header size: %d): file may be corrupted or a Git LFS pointer", p, n)
}
b := bytes.NewBuffer(make([]byte, 0, n)) b := bytes.NewBuffer(make([]byte, 0, n))
if _, err = io.CopyN(b, f, n); err != nil { if _, err = io.CopyN(b, f, n); err != nil {
return nil, err return nil, err