OpenCompass/configs/models/hf_llama/hf_llama3_1_8b_instruct.py

23 lines
615 B
Python
Raw Normal View History

from opencompass.models import HuggingFacewithChatTemplate
api_meta_template = dict(
round=[
dict(role='HUMAN', api_role='HUMAN'),
dict(role='BOT', api_role='BOT', generate=True),
],
reserved_roles=[dict(role='SYSTEM', api_role='SYSTEM')],
)
models = [
dict(
type=HuggingFacewithChatTemplate,
abbr='llama-3_1-8b-instruct-hf',
path='meta-llama/Meta-Llama-3.1-8B-Instruct',
max_out_len=1024,
batch_size=8,
run_cfg=dict(num_gpus=1),
stop_words=['<|end_of_text|>', '<|eot_id|>'],
meta_template=api_meta_template,
)
]