Commit 0d8e8e14 authored by zhuwenwen's avatar zhuwenwen
Browse files

update vllm and lmdeploy configs

parent b3797475
...@@ -69,9 +69,9 @@ python tools/list_configs.py llama mmlu ...@@ -69,9 +69,9 @@ python tools/list_configs.py llama mmlu
1、使用vllm推理验证 1、使用vllm推理验证
```shell ```shell
python run.py configs/eval_llama2_vllm.py python run.py configs/vllm/eval_llama2_vllm.py
``` ```
其它模型使用参考`configs/eval_xxx_vllm.py` 其它模型使用参考`configs/vllm/eval_xxx_vllm.py`
2、使用lmdeploy推理验证 2、使用lmdeploy推理验证
```shell ```shell
...@@ -85,7 +85,7 @@ source /opt/dtk-24.04/cuda/env.sh ...@@ -85,7 +85,7 @@ source /opt/dtk-24.04/cuda/env.sh
#进入opencompass进行评测 #进入opencompass进行评测
cd opencompass cd opencompass
#fp16精度评测方法 #fp16精度评测方法
python run.py configs/eval_llama2_lmdeploy.py python run.py configs/lmdeploy/eval_llama2_lmdeploy.py
#awq int4 评测方法 #awq int4 评测方法
#首先需要转换awqInt4模型 #首先需要转换awqInt4模型
#model_name:模型名字如 llama2 qwen-7b #model_name:模型名字如 llama2 qwen-7b
...@@ -93,13 +93,15 @@ python run.py configs/eval_llama2_lmdeploy.py ...@@ -93,13 +93,15 @@ python run.py configs/eval_llama2_lmdeploy.py
#awq_lmdeploymodel_path:生成的lmdeploy格式 awq模型路径 #awq_lmdeploymodel_path:生成的lmdeploy格式 awq模型路径
lmdeploy convert ${model_name} ${awq_modelpath} --model-format awq --group-size 128 --dst-path ${awq_lmdeploymodel_path} lmdeploy convert ${model_name} ${awq_modelpath} --model-format awq --group-size 128 --dst-path ${awq_lmdeploymodel_path}
#将eval_llama2_lmdeploy.py中的Llama-2-7b-hf 替换为转换好的awq_lmdeploymodel_path #将eval_llama2_lmdeploy.py中的Llama-2-7b-hf 替换为转换好的awq_lmdeploymodel_path
python run.py configs/eval_llama2_lmdeploy.py python run.py configs/lmdeploy/eval_llama2_lmdeploy.py
``` ```
其它模型使用参考`configs/lmdeploy/eval_xxx_lmdeploy.py`
3、使用tgi推理验证 3、使用tgi推理验证
```shell ```shell
python run.py configs/eval_llama2_tgi.py python run.py configs/tgi/eval_llama2_tgi.py
``` ```
其它模型使用参考`configs/tgi/eval_xxx_tgi.py`
参数说明: 参数说明:
(1)数据集配置参数 (1)数据集配置参数
......
...@@ -8,4 +8,4 @@ with read_base(): ...@@ -8,4 +8,4 @@ with read_base():
from .datasets.mmlu.mmlu_gen import mmlu_datasets from .datasets.mmlu.mmlu_gen import mmlu_datasets
from .models.alaya.alaya import models from .models.alaya.alaya import models
datasets = [*bbh_datasets, *ceval_datasets, *cmmlu_datasets, *agieval_datasets, *mmlu_datasets] datasets = [*bbh_datasets, *ceval_datasets, *cmmlu_datasets, *agieval_datasets, *mmlu_datasets]
\ No newline at end of file
from mmengine.config import read_base from mmengine.config import read_base
from opencompass.models.turbomind import TurboMindModel from opencompass.models.turbomind import TurboMindModel
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), []) datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), [])
work_dir = './outputs/llama2-chat/' work_dir = './outputs/llama2-chat/'
llama_chat_meta_template = dict( llama_chat_meta_template = dict(
round=[ round=[
dict(role="HUMAN", begin='[INST] ', end=' [/INST]'), dict(role="HUMAN", begin='[INST] ', end=' [/INST]'),
dict(role="BOT", begin=' ', end=' ', generate=True), dict(role="BOT", begin=' ', end=' ', generate=True),
], ],
) )
models = [ models = [
dict( dict(
type=TurboMindModel, type=TurboMindModel,
abbr='llama-2-7b-chat-hf-lmdeploy', abbr='llama-2-7b-chat-hf-lmdeploy',
path="Llama-2-7b-chat-hf", path="Llama-2-7b-chat-hf",
meta_template=llama_chat_meta_template, meta_template=llama_chat_meta_template,
engine_config=dict(session_len=4096, engine_config=dict(session_len=4096,
max_batch_size=32), max_batch_size=32),
gen_config=dict(top_k=1, gen_config=dict(top_k=1,
top_p=0.8, top_p=0.8,
temperature=1.0, temperature=1.0,
max_new_tokens=100), max_new_tokens=100),
max_out_len=100, max_out_len=100,
max_seq_len=2048, max_seq_len=2048,
batch_size=1, batch_size=1,
concurrency=1, concurrency=1,
run_cfg=dict(num_gpus=1, num_procs=1), run_cfg=dict(num_gpus=1, num_procs=1),
end_str='[INST]', end_str='[INST]',
) )
] ]
\ No newline at end of file
...@@ -2,9 +2,9 @@ from mmengine.config import read_base ...@@ -2,9 +2,9 @@ from mmengine.config import read_base
from opencompass.models.turbomind import TurboMindModel from opencompass.models.turbomind import TurboMindModel
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), []) datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), [])
work_dir = './outputs/llama2/' work_dir = './outputs/llama2/'
......
...@@ -2,9 +2,9 @@ from mmengine.config import read_base ...@@ -2,9 +2,9 @@ from mmengine.config import read_base
from opencompass.models.turbomind import TurboMindModel from opencompass.models.turbomind import TurboMindModel
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), []) datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), [])
work_dir = './outputs/qwen-chat/' work_dir = './outputs/qwen-chat/'
......
...@@ -2,9 +2,9 @@ from mmengine.config import read_base ...@@ -2,9 +2,9 @@ from mmengine.config import read_base
from opencompass.models.turbomind import TurboMindModel from opencompass.models.turbomind import TurboMindModel
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), []) datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), [])
work_dir = './outputs/qwen/' work_dir = './outputs/qwen/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .datasets.ceval.ceval_gen_5f30c7 import ceval_datasets from ..datasets.ceval.ceval_gen_5f30c7 import ceval_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/baichuan2-chat/' work_dir = './outputs/baichuan2-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/chatglm3/' work_dir = './outputs/chatglm3/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/llama2-chat/' work_dir = './outputs/llama2-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/llama2-chat/' work_dir = './outputs/llama2-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/llama2/' work_dir = './outputs/llama2/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/llama3-instruct/' work_dir = './outputs/llama3-instruct/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .datasets.ceval.ceval_gen_5f30c7 import ceval_datasets from ..datasets.ceval.ceval_gen_5f30c7 import ceval_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/qwen1.5-chat/' work_dir = './outputs/qwen1.5-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .datasets.ceval.ceval_gen_5f30c7 import ceval_datasets from ..datasets.ceval.ceval_gen_5f30c7 import ceval_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/qwen1.5-int4-chat/' work_dir = './outputs/qwen1.5-int4-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .datasets.ceval.ceval_gen_5f30c7 import ceval_datasets from ..datasets.ceval.ceval_gen_5f30c7 import ceval_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/qwen1.5/' work_dir = './outputs/qwen1.5/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/qwen-chat/' work_dir = './outputs/qwen-chat/'
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets from ..datasets.ARC_c.ARC_c_gen_1e0de5 import ARC_c_datasets
from .datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets from ..datasets.ARC_e.ARC_e_gen_1e0de5 import ARC_e_datasets
from .summarizers.example import summarizer from ..summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], []) datasets = sum([v for k, v in locals().items() if k.endswith("_datasets") or k == 'datasets'], [])
work_dir = './outputs/qwen-int4-chat/' work_dir = './outputs/qwen-int4-chat/'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment