首页
/ Docling项目模型权重转换为Safetensors格式的技术实践

Docling项目模型权重转换为Safetensors格式的技术实践

2025-05-06 08:49:39作者:翟江哲Frasier

在自然语言处理领域,模型权重的存储格式对系统性能和安全性有着重要影响。DS4SD团队近期在Docling项目中完成了一项重要技术升级——将原有的pickle格式模型权重全面迁移至Safetensors格式。这一技术改进为项目带来了显著的安全性和性能提升。

技术背景

传统上,PyTorch框架通常使用pickle或torch.jit格式保存模型权重。然而,pickle格式存在严重的安全隐患,因为它可以执行任意代码,这使得从不可信来源加载模型变得危险。此外,pickle格式在跨平台兼容性和加载效率方面也存在不足。

Safetensors是HuggingFace推出的一种新型张量存储格式,具有以下核心优势:

  1. 安全性:不会执行任意代码,从根本上杜绝了恶意代码注入的风险
  2. 高效性:支持零拷贝加载,显著提升模型加载速度
  3. 跨平台:与多种编程语言和硬件平台兼容
  4. 轻量级:文件体积通常小于传统格式

技术实现

Docling项目团队在v2.12.0版本中完成了这一重要升级。转换过程主要涉及以下技术环节:

  1. 模型加载重构:重写了模型加载逻辑,使用safetensors库替代原有的torch.load方法
  2. 格式转换工具:开发了自动化脚本,将现有pickle格式权重批量转换为safetensors格式
  3. 兼容性处理:确保新格式与项目原有接口完全兼容,不影响上层应用
  4. 性能优化:利用safetensors的零拷贝特性优化加载速度

实际效果

根据用户反馈,这一改进带来了显著的实际效益。以PDF文档处理为例:

  • 处理时间从原来的14秒大幅降低至几秒级别
  • 内存占用减少约30%
  • 安全性得到本质提升,消除了潜在的安全隐患

特别值得注意的是,这一改进不仅提升了性能,还保留了Docling原有的优势——能够完美保持文档的原始格式信息,这在处理复杂排版文档时尤为重要。

未来展望

Safetensors格式的采用为Docling项目奠定了更好的技术基础。团队计划在此基础上进一步优化:

  1. 探索量化技术,进一步减小模型体积
  2. 研究异构计算支持,充分利用GPU等加速硬件
  3. 优化多模型并行加载机制

这一技术升级体现了DS4SD团队对项目质量和用户体验的持续追求,也为同类项目的技术选型提供了有价值的参考案例。

登录后查看全文

热门内容推荐

项目优选

收起
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
604
424
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
51
15
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
128
209
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
90
146
cherry-studiocherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
479
39
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
106
255
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
299
1.03 K
MateChatMateChat
前端智能化场景解决方案UI库,轻松构建你的AI应用,我们将持续完善更新,欢迎你的使用与建议。 官网地址:https://matechat.gitcode.com
693
92
markdown4cjmarkdown4cj
一个markdown解析和展示的库
Cangjie
33
4
JeecgBootJeecgBoot
🔥企业级低代码平台集成了AI应用平台,帮助企业快速实现低代码开发和构建AI应用!前后端分离架构 SpringBoot,SpringCloud、Mybatis,Ant Design4、 Vue3.0、TS+vite!强大的代码生成器让前后端代码一键生成,无需写任何代码! 引领AI低代码开发模式: AI生成->OnlineCoding-> 代码生成-> 手工MERGE,显著的提高效率,又不失灵活~
Java
96
17