eval_demo.py 636 Bytes
Newer Older
yuzhaohui's avatar
yuzhaohui committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from opencompass.models import HuggingFaceCausalLM

with read_base():
    from .datasets.piqa.piqa_ppl import piqa_datasets 

datasets = piqa_datasets

models = [
    # OPT-1.3b
    dict(
       type=HuggingFaceCausalLM,
       path='facebook/opt-1.3b',
       tokenizer_path='facebook/opt-1.3b',
       tokenizer_kwargs=dict(
           padding_side='left',
           truncation_side='left',
           proxies=None,
           trust_remote_code=True,
       ),
       max_out_len=100,
       max_seq_len=2048,
       batch_size=16,
       model_kwargs=dict(device_map='auto'),
       run_cfg=dict(num_gpus=1, num_procs=1),
    )
]