"...pipelines/hunyuan_image/pipeline_hunyuanimage.py" did not exist on "be2070991f1b916977020c45ecdfec225de21862"
hf_llama_7b.py 677 Bytes
Newer Older
cky's avatar
cky committed
1
2
3
4
5
6
from opencompass.models import HuggingFaceCausalLM

models = [
    # LLaMA 7B
    dict(
        type=HuggingFaceCausalLM,
Leymore's avatar
Leymore committed
7
        abbr='llama-7b-hf',
Tong Gao's avatar
Tong Gao committed
8
9
        path="huggyllama/llama-7b",
        tokenizer_path='huggyllama/llama-7b',
cky's avatar
cky committed
10
11
12
13
14
15
16
17
        tokenizer_kwargs=dict(padding_side='left',
                              truncation_side='left',
                              use_fast=False,
                              ),
        max_out_len=100,
        max_seq_len=2048,
        batch_size=8,
        model_kwargs=dict(device_map='auto'),
Leymore's avatar
Leymore committed
18
19
        batch_padding=False, # if false, inference with for-loop without batch padding
        run_cfg=dict(num_gpus=1, num_procs=1),
cky's avatar
cky committed
20
21
    )
]