Docling项目模型权重转换为Safetensors格式的技术实践
2025-05-06 08:49:39作者:翟江哲Frasier
在自然语言处理领域,模型权重的存储格式对系统性能和安全性有着重要影响。DS4SD团队近期在Docling项目中完成了一项重要技术升级——将原有的pickle格式模型权重全面迁移至Safetensors格式。这一技术改进为项目带来了显著的安全性和性能提升。
技术背景
传统上,PyTorch框架通常使用pickle或torch.jit格式保存模型权重。然而,pickle格式存在严重的安全隐患,因为它可以执行任意代码,这使得从不可信来源加载模型变得危险。此外,pickle格式在跨平台兼容性和加载效率方面也存在不足。
Safetensors是HuggingFace推出的一种新型张量存储格式,具有以下核心优势:
- 安全性:不会执行任意代码,从根本上杜绝了恶意代码注入的风险
- 高效性:支持零拷贝加载,显著提升模型加载速度
- 跨平台:与多种编程语言和硬件平台兼容
- 轻量级:文件体积通常小于传统格式
技术实现
Docling项目团队在v2.12.0版本中完成了这一重要升级。转换过程主要涉及以下技术环节:
- 模型加载重构:重写了模型加载逻辑,使用safetensors库替代原有的torch.load方法
- 格式转换工具:开发了自动化脚本,将现有pickle格式权重批量转换为safetensors格式
- 兼容性处理:确保新格式与项目原有接口完全兼容,不影响上层应用
- 性能优化:利用safetensors的零拷贝特性优化加载速度
实际效果
根据用户反馈,这一改进带来了显著的实际效益。以PDF文档处理为例:
- 处理时间从原来的14秒大幅降低至几秒级别
- 内存占用减少约30%
- 安全性得到本质提升,消除了潜在的安全隐患
特别值得注意的是,这一改进不仅提升了性能,还保留了Docling原有的优势——能够完美保持文档的原始格式信息,这在处理复杂排版文档时尤为重要。
未来展望
Safetensors格式的采用为Docling项目奠定了更好的技术基础。团队计划在此基础上进一步优化:
- 探索量化技术,进一步减小模型体积
- 研究异构计算支持,充分利用GPU等加速硬件
- 优化多模型并行加载机制
这一技术升级体现了DS4SD团队对项目质量和用户体验的持续追求,也为同类项目的技术选型提供了有价值的参考案例。
登录后查看全文
热门项目推荐
cherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端TypeScript039RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统Vue0424arkanalyzer
方舟分析器:面向ArkTS语言的静态程序分析框架TypeScript041GitCode百大开源项目
GitCode百大计划旨在表彰GitCode平台上积极推动项目社区化,拥有广泛影响力的G-Star项目,入选项目不仅代表了GitCode开源生态的蓬勃发展,也反映了当下开源行业的发展趋势。03PowerWechat
PowerWechat是一款基于WeChat SDK for Golang,支持小程序、微信支付、企业微信、公众号等全微信生态Go01openGauss-server
openGauss kernel ~ openGauss is an open source relational database management systemC++0146
热门内容推荐
最新内容推荐
Visual-RFT项目中模型路径差异的技术解析 Microcks在OpenShift上部署Keycloak PostgreSQL的权限问题解析 Beyla项目中的HTTP2连接检测问题解析 RaspberryMatic项目中HmIP-BWTH温控器假期模式设置问题分析 Lets-Plot 库中条形图标签在坐标轴反转时的定位问题解析 BedrockConnect项目版本兼容性问题解析与解决方案 LiquidJS 10.21.0版本新增数组过滤功能解析 Mink项目中Selenium驱动切换iframe的兼容性问题分析 Lichess移动端盲棋模式字符串优化解析 sbctl验证功能JSON输出问题解析
项目优选
收起

🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
604
424

🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
51
15

React Native鸿蒙化仓库
C++
128
209

openGauss kernel ~ openGauss is an open source relational database management system
C++
90
146

🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
479
39

旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
106
255

本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
299
1.03 K

前端智能化场景解决方案UI库,轻松构建你的AI应用,我们将持续完善更新,欢迎你的使用与建议。
官网地址:https://matechat.gitcode.com
693
92

一个markdown解析和展示的库
Cangjie
33
4

🔥企业级低代码平台集成了AI应用平台,帮助企业快速实现低代码开发和构建AI应用!前后端分离架构 SpringBoot,SpringCloud、Mybatis,Ant Design4、 Vue3.0、TS+vite!强大的代码生成器让前后端代码一键生成,无需写任何代码! 引领AI低代码开发模式: AI生成->OnlineCoding-> 代码生成-> 手工MERGE,显著的提高效率,又不失灵活~
Java
96
17