OpenCompass/opencompass/configs/models/bailing_api/bailing-lite-20250220.py

29 lines
754 B
Python
Raw Normal View History

from opencompass.models import BailingAPI
api_meta_template = dict(
round=[
dict(role='HUMAN', api_role='HUMAN'),
dict(role='BOT', api_role='BOT', generate=False),
],
reserved_roles=[dict(role='SYSTEM', api_role='SYSTEM')],
)
models = [
dict(
2025-04-21 10:37:00 +08:00
path='Bailing-Lite-20250220',
token='', # set your key here or in environment variable BAILING_API_KEY
url='https://bailingchat.alipay.com/chat/completions',
type=BailingAPI,
meta_template=api_meta_template,
2025-04-21 10:37:00 +08:00
max_out_len=16384,
batch_size=1,
generation_kwargs={
'temperature': 0.01,
'top_p': 1.0,
'top_k': -1,
'n': 1,
'logprobs': 1,
},
),
]