1
2
1
AI人工智能
orion-14b-chat 模型推理乱码
2024-05-15 12:37:08
145

部署环境: 本地conda安装,dtk 24.04 ,编译器 mpi :compiler/devtoolset/7.3.1 mpi/hpcx/2.11.0-gcc-7.3.1
安装的移植包:

torch 2.1.0+git00661e0.abi0.dtk2404
paddlepaddle-rocm 2.5.2

flash-attn 2.0.4+82379d7.abi0.dtk2404.torch2.1
推理过程中出现部分乱码情况,一般是什么原因呀,怎么分析解决?
使用源码:https://huggingface.co/OrionStarAI/Orion-14B-Chat

attachmentId-image.png

 

全部评论(3)
只看楼主
P
panhuiwen
1楼

这个模型我们暂时还未验证,待验证后会上传到modelzoo。

看到您用的是paddle+fa,请问推理框架用的是paddle吗?我看到orion模型官方推荐的是vllm

初期的排查建议您优先提升数据精度至FP32看看是否任然存在乱码问题。

2024/5/15
G
GEEK
回复 @panhuiwen
直接transformer推理都乱码,尝试指定编码也无效,提升数据精度至FP32会爆显存;使用Z100 16G显存*4卡节点运行。有没有办法确认 是否存在哪层网络溢出或者精度下降之类的?
2024/5/20
G
GEEK
回复 @panhuiwen
升级到dtk24.04 最新的torch环境后,测试乱码正常,感谢支持
2024/5/23
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据