Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
hf_opt_125m.py 262 Bytes
Newer Older
Fengzhe Zhou's avatar
[Feature] Add huggingface apply_chat_template (#1098)  
Fengzhe Zhou committed
1 
from opencompass.models import HuggingFaceBaseModel
Tong Gao's avatar
[Docs] Update docs for new entry script (#246)  
Tong Gao committed
2
Fengzhe Zhou's avatar
[Feature] Add huggingface apply_chat_template (#1098)  
Fengzhe Zhou committed
3
4
5
6
7
8
9
10 
models = [
    dict(
        type=HuggingFaceBaseModel,
        abbr='opt-125m-hf',
        path='facebook/opt-125m',
        max_out_len=1024,
        batch_size=64,
        run_cfg=dict(num_gpus=1),
Tong Gao's avatar
[Docs] Update docs for new entry script (#246)  
Tong Gao committed
11 
    )
Fengzhe Zhou's avatar
[Feature] Add huggingface apply_chat_template (#1098)  
Fengzhe Zhou committed
12 
]