2
1
框架组件
通过vllm执行推理报错
2024-10-15 09:44:34
197

根据光源的部署文档光源 Find source, find chance. (sourcefind.cn)
最后执行推理的命令:

python ./inference_vllm/Qwen2_72B_inference.py

报错如下图:ValueError: gptq_marlin quantization is currently not supported in ROCm
注:docker镜像为:docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu22.04-dtk24.04.2-py3.10
镜像里面安装的具体pip依赖,请看附件。

全部评论(2)
只看楼主
Z
zhuwenwen
2楼
提供下运行的py文件,若使用fp16模型,则指定quantization=None,若使用gptq模型,需要添加quantization=“gptq”
2024/10/21
P
panhuiwen
1楼
gptq-marlin暂时还不支持
2024/10/16
没有更多内容了
--
楼主
Level.6
0 帖子
楼主热帖
暂无数据