OpenCompass/configs/models/others/hf_command_r_plus.py

13 lines
295 B
Python
Raw Permalink Normal View History

from opencompass.models import HuggingFacewithChatTemplate
2024-04-19 20:49:46 +08:00
models = [
dict(
type=HuggingFacewithChatTemplate,
2024-04-19 20:49:46 +08:00
abbr='command-r-plus-hf',
path='CohereForAI/c4ai-command-r-plus',
max_out_len=1024,
2024-04-19 20:49:46 +08:00
batch_size=8,
run_cfg=dict(num_gpus=8),
2024-04-19 20:49:46 +08:00
)
]