会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
将者,智、信、仁、勇、严也。
Hi,我是李智华,华为-安全AI算法专家,欢迎来到安全攻防对抗的有趣世界。
博客园
首页
新随笔
联系
订阅
管理
2024年6月29日
python BGE 模型转换为onnx给java调用
摘要: 最近在做RAG,因为涉及embedding计算,用到了BAAI BGE小模型,但是模型是给python调用的,需要转换为onnx格式给java使用。所以有了下面的探索: python代码: import torch from transformers import AutoTokenizer, Au
阅读全文
posted @ 2024-06-29 11:39 bonelee
阅读(153)
评论(1)
推荐(0)
编辑
公告