OpenCompass/configs/models/qwen2_5/hf_qwen2_5_3b_instruct.py
2024-09-19 16:16:07 +08:00

13 lines
293 B
Python

from opencompass.models import HuggingFacewithChatTemplate
models = [
dict(
type=HuggingFacewithChatTemplate,
abbr='qwen2.5-3b-instruct-hf',
path='Qwen/Qwen2.5-3B-Instruct',
max_out_len=4096,
batch_size=8,
run_cfg=dict(num_gpus=1),
)
]