open-compass / open-compass/opencompass
[Feature] 请问使用API评测如何支持自定义数据集?
Open
@acylam is already working on this.
Since Nov 28, 2024.
- Dominant language
- Python
- Stars
- 7.5k
- Forks
- 869
- Avg merge
- 17h 52m
- Merged PRs (30d)
- 13
Description
Describe the feature
command:
(opencompass) root@node1:~/user/OpenCompass/opencompass# python run.py configs/vllm-glm4-9b-chat-custom.py
vllm-glm4-9b-chat-custom.py
from opencompass.models import OpenAISDK
from mmengine.config import read_base
with read_base():
from .datasets.cmmlu.cmmlu_gen import cmmlu_datasets
datasets = []
datasets = cmmlu_datasets
api_meta_template = dict(
round=[
dict(role='HUMAN', api_role='HUMAN'),
dict(role='BOT', api_role='BOT', generate=True),
],
reserved_roles=[dict(role='SYSTEM', api_role='SYSTEM')],
)
models = [
dict(
abbr='glm-4-9b-chat-vllm-API',
type=OpenAISDK,
key='EMPTY',
openai_api_base='http://localhost:port/v1',
path='glm-4-9b-chat',
tokenizer_path='/root/user/models/glm-4-9b-chat',
rpm_verbose=True,
meta_template=api_meta_template,
query_per_second=10,
max_out_len=1024,
max_seq_len=4096,
temperature=0.01,
batch_size=8,
retry=3,
)
]
想在不更改OpenCompass代码的情况下实现,但是文档示例只有以下用法:
python run.py \
--models hf_llama2_7b \
--custom-dataset-path xxx/test_qa.jsonl \
--custom-dataset-data-type qa \
--custom-dataset-infer-method gen
Will you implement it?
- I would like to implement this feature and create a PR!
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.