File size: 727 Bytes
256a159 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from mmengine.config import read_base
with read_base():
from .datasets.ceval.ceval_clean_ppl import ceval_datasets
from .datasets.mmlu.mmlu_clean_ppl import mmlu_datasets
from .datasets.hellaswag.hellaswag_clean_ppl import hellaswag_datasets
from .datasets.ARC_c.ARC_c_clean_ppl import ARC_c_datasets
from .models.yi.hf_yi_6b import models as hf_yi_6b_model
from .models.qwen.hf_qwen_7b import models as hf_qwen_7b_model
from .models.hf_llama.hf_llama2_7b import models as hf_llama2_7b_model
from .summarizers.contamination import summarizer
datasets = [*ceval_datasets, *mmlu_datasets, *hellaswag_datasets, *ARC_c_datasets]
models = [*hf_yi_6b_model, *hf_qwen_7b_model, *hf_llama2_7b_model]
|