From 167457f1ffb142117a75f9fdb4d4ef41aee25538 Mon Sep 17 00:00:00 2001 From: TwT-JD <163280725+TwT-JD@users.noreply.github.com> Date: Wed, 13 Mar 2024 17:38:53 +0800 Subject: [PATCH] Add files via upload --- configs/models/my_api/my_api.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 configs/models/my_api/my_api.py diff --git a/configs/models/my_api/my_api.py b/configs/models/my_api/my_api.py new file mode 100644 index 00000000..eedc427a --- /dev/null +++ b/configs/models/my_api/my_api.py @@ -0,0 +1,21 @@ +from opencompass.models.my_api import MyAPIModel +api_meta_template = dict( + round=[ + dict(role='HUMAN', api_role='HUMAN'), + dict(role='BOT', api_role='BOT', generate=True), + ], +) +models = [ + dict( + type=MyAPIModel, + abbr='my_api', + path='my_api', + url='https://api-opencompass.jd.com/testing', + api_key = 'w8QA7LSXQG1q9Tc1A0X3P8PWXMkmyuPSCPtRSCg9NtM95dBlpO', + meta_template=api_meta_template, + max_out_len=100, + max_seq_len=2048, + batch_size=8, + run_cfg=dict(num_gpus=1, num_procs=1), + ) +] \ No newline at end of file