2
9
数学库
标题
2024-10-11 14:33:25
178

vllm部署baichuan2-7b awq w4a16量化后的模型报错。

DCU 加速卡驱动程序:rock-5.7.1-6.2.18-V1.1.2.aio.run
DCU 加速卡软件工具包:DTK-24.04.2

加速卡: K100_AI

vllm:0.5.0+das.opt1.dtk24042
torch: 2.1.0+das.opt1.dtk24042
其他软件根据此依赖。环境部署qwen1.5 qwen2,以及其awq量化后的模型都可以。部署baichuan2-7b基础模型也可以。但部署baichuan2-7b awq量化后的模型则不行,报错信息为:
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/vllm/model_executor/models/baichuan.py", line 248, in forward
hidden_states = self.self_attn(
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/vllm/model_executor/models/baichuan.py", line 196, in forward
attn_output = self.attn(q, k, v, kv_cache, attn_metadata)
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/vllm/attention/layer.py", line 89, in forward
return self.impl.forward(query, key, value, kv_cache, attn_metadata,
File "/home/xxx/.conda/envs/vllm/lib/python3.10/site-packages/vllm/attention/backends/rocm_flash_attn.py", line 297, in forward
query = query.view(-1, self.num_heads, self.head_size)
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

全部评论(4)
只看楼主
Z
zhuwenwen
3楼
das1.3发布的vllm0.5.0将会支持baichuan的awq量化
2024/10/21
网
网友po20TW
回复 @zhuwenwen
啥时候vllm升级到0.6.0以上啊
2024/10/22
W
wangwei
2楼
目前baichaun系列的模型还不支持awq量化
2024/10/14
A
ANYA
1楼
已经在验证中,请您等待后续的消息
2024/10/11
没有更多内容了
--
楼主
Level.3
0 帖子
楼主热帖
暂无数据