Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
internlm_7b.py 340 Bytes
Newer Older
gowithme's avatar
[Feature] Support intern lanuage model (#51)  
gowithme committed
1
2
3
4
5
6 
from opencompass.models import InternLM


models = [
    dict(
        type=InternLM,
Fengzhe Zhou's avatar
[Format] Add config lints (#892)  
Fengzhe Zhou committed
7 
        path='./internData/',
gowithme's avatar
[Feature] Support intern lanuage model (#51)  
gowithme committed
8 
        tokenizer_path='./internData/V7.model',
Fengzhe Zhou's avatar
[Format] Add config lints (#892)  
Fengzhe Zhou committed
9 
        model_config='./internData/model_config.py',
gowithme's avatar
[Feature] Support intern lanuage model (#51)  
gowithme committed
10
11
12
13
14 
        max_out_len=100,
        max_seq_len=2048,
        batch_size=16,
        run_cfg=dict(num_gpus=1, num_procs=1))
]