OpenCompass/configs/models/qwen/vllm_qwen1_5_14b_chat.py

15 lines
374 B
Python
Raw Normal View History

from opencompass.models import VLLMwithChatTemplate
2024-03-01 17:31:51 +08:00
models = [
dict(
type=VLLMwithChatTemplate,
2024-03-01 17:31:51 +08:00
abbr='qwen1.5-14b-chat-vllm',
2024-05-14 15:35:58 +08:00
path='Qwen/Qwen1.5-14B-Chat',
model_kwargs=dict(tensor_parallel_size=1),
max_out_len=1024,
batch_size=16,
2024-03-01 17:31:51 +08:00
generation_kwargs=dict(temperature=0),
run_cfg=dict(num_gpus=1),
2024-03-01 17:31:51 +08:00
)
]