internlm_7b.py

from opencompass.models import InternLM


models = [
    dict(
        type=InternLM,
        path="./internData/",
        tokenizer_path='./internData/V7.model',
        model_config="./internData/model_config.py",
        max_out_len=100,
        max_seq_len=2048,
        batch_size=16,
        run_cfg=dict(num_gpus=1, num_procs=1))
]