open-compass / open-compass/opencompass
[Feature] 请问使用vllm评测时怎么实现类似HF多卡数据并行？
Open
@tonysy is already working on this.
Since Mar 26, 2024.
- Dominant language
- Python
- Stars
- 7.5k
- Forks
- 869
- Avg merge
- 17h 52m
- Merged PRs (30d)
- 13
Description
描述该功能
我在评测时的模型type 为vllm，参数如下：
但是显卡占用只使用了一张卡来评测任务
我想让任务划分为几份分别在8张卡上评测，这种功能可以添加吗？还是说可以实现，麻烦解答一下。非常感激！
类似我如果设定为模型type为HF的话，会自动达到这种效果。
是否希望自己实现该功能？
- 我希望自己来实现这一功能，并向 OpenCompass 贡献代码！
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.