forked from open-compass/opencompass
-
Notifications
You must be signed in to change notification settings - Fork 0
/
eval_hf_llama2.py
18 lines (16 loc) · 1.21 KB
/
eval_hf_llama2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from mmengine.config import read_base
with read_base():
from opencompass.configs.datasets.mmlu.mmlu_ppl_ac766d import mmlu_datasets
from opencompass.configs.datasets.triviaqa.triviaqa_wiki_gen_d18bf4 import triviaqa_datasets
from opencompass.configs.datasets.nq.nq_open_gen_e93f8a import nq_datasets
from opencompass.configs.datasets.gsm8k.gsm8k_gen_3309bd import gsm8k_datasets
from opencompass.configs.datasets.humaneval.deprecated_humaneval_gen_a82cae import humaneval_datasets
from opencompass.configs.datasets.agieval.agieval_mixed_713d14 import agieval_datasets
from opencompass.configs.datasets.SuperGLUE_BoolQ.SuperGLUE_BoolQ_ppl_314797 import BoolQ_datasets
from opencompass.configs.datasets.hellaswag.hellaswag_ppl_a6e128 import hellaswag_datasets
from opencompass.configs.datasets.obqa.obqa_ppl_6aac9e import obqa_datasets
from opencompass.configs.datasets.winogrande.winogrande_ll_c5cf57 import winogrande_datasets
from opencompass.configs.models.hf_llama.hf_llama2_7b import models
from opencompass.configs.summarizers.example import summarizer
datasets = sum([v for k, v in locals().items() if k.endswith('_datasets') or k == 'datasets'], [])
work_dir = './outputs/llama2/'