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

View File

@ -37,6 +37,10 @@ func parseSafetensors(fsys fs.FS, replacer *strings.Replacer, ps ...string) ([]T
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))
if _, err = io.CopyN(b, f, n); err != nil {
return nil, err