hf_internlm2_7b.py 270 Bytes
Newer Older
1
from opencompass.models import HuggingFaceBaseModel
2
3
4
5


models = [
    dict(
6
        type=HuggingFaceBaseModel,
7
        abbr='internlm2-7b-hf',
8
        path='internlm/internlm2-7b',
9
        max_out_len=1024,
10
        batch_size=8,
11
        run_cfg=dict(num_gpus=1),
12
13
    )
]