Compare commits

...

1 Commits

Author SHA1 Message Date
ParthSareen
5d3eeb43c0 convert: check file size for safetensors to warn for improper conversion 2025-12-10 17:58:16 -08:00

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