Spaces:
Runtime error
Runtime error
fanqiNO1
commited on
Commit
·
1e5d6ca
1
Parent(s):
0082e2e
[Feature] Support SiliconFlow API (#169)
Browse files- mindsearch/agent/models.py +17 -0
mindsearch/agent/models.py
CHANGED
@@ -58,3 +58,20 @@ qwen = dict(type=GPTAPI,
|
|
58 |
max_new_tokens=4096,
|
59 |
repetition_penalty=1.02,
|
60 |
stop_words=['<|im_end|>'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
max_new_tokens=4096,
|
59 |
repetition_penalty=1.02,
|
60 |
stop_words=['<|im_end|>'])
|
61 |
+
|
62 |
+
internlm_silicon = dict(type=GPTAPI,
|
63 |
+
model_type='internlm/internlm2_5-7b-chat',
|
64 |
+
key=os.environ.get('SILICON_API_KEY', 'YOUR SILICON API KEY'),
|
65 |
+
openai_api_base='https://api.siliconflow.cn/v1/chat/completions',
|
66 |
+
meta_template=[
|
67 |
+
dict(role='system', api_role='system'),
|
68 |
+
dict(role='user', api_role='user'),
|
69 |
+
dict(role='assistant', api_role='assistant'),
|
70 |
+
dict(role='environment', api_role='system')
|
71 |
+
],
|
72 |
+
top_p=0.8,
|
73 |
+
top_k=1,
|
74 |
+
temperature=0,
|
75 |
+
max_new_tokens=8192,
|
76 |
+
repetition_penalty=1.02,
|
77 |
+
stop_words=['<|im_end|>'])
|