返回提交历史
Modified
g4f/models.py
+16
-9
XFEstudio/gpt4free
Update DeepInfra models
2e171fc2
代码差异
1 个文件
+16
-9
@@ -127,7 +127,7 @@ llama2_70b = Model(
127
127
codellama_34b_instruct = Model(
128
128
name = "codellama/CodeLlama-34b-Instruct-hf",
129
129
base_provider = "meta",
130
best_provider = RetryProvider([HuggingChat, DeepInfra])
130
best_provider = HuggingChat
131
131
)
132
132
133
133
codellama_70b_instruct = Model(
@@ -146,9 +146,21 @@ mixtral_8x7b = Model(
146
146
mistral_7b = Model(
147
147
name = "mistralai/Mistral-7B-Instruct-v0.1",
148
148
base_provider = "huggingface",
149
best_provider = RetryProvider([DeepInfra, HuggingChat, HuggingFace, PerplexityLabs])
149
best_provider = RetryProvider([HuggingChat, HuggingFace, PerplexityLabs])
150
)
151
152
mistral_7b_v02 = Model(
153
name = "mistralai/Mistral-7B-Instruct-v0.2",
154
base_provider = "huggingface",
155
best_provider = DeepInfra
150
156
)
151
157
158
# mixtral_8x22b = Model(
159
# name = "mistralai/Mixtral-8x22B-v0.1",
160
# base_provider = "huggingface",
161
# best_provider = DeepInfra
162
# )
163
152
164
# Misc models
153
165
dolphin_mixtral_8x7b = Model(
154
166
name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b",
@@ -168,12 +180,6 @@ airoboros_70b = Model(
168
180
best_provider = DeepInfra
169
181
)
170
182
171
airoboros_l2_70b = Model(
172
name = "jondurbin/airoboros-l2-70b-gpt4-1.4.1",
173
base_provider = "huggingface",
174
best_provider = DeepInfra
175
)
176
177
183
openchat_35 = Model(
178
184
name = "openchat/openchat_3.5",
179
185
base_provider = "huggingface",
@@ -290,10 +296,11 @@ class ModelUtils:
290
296
291
297
'mixtral-8x7b': mixtral_8x7b,
292
298
'mistral-7b': mistral_7b,
299
'mistral-7b-v02': mistral_7b_v02,
300
# 'mixtral-8x22b': mixtral_8x22b,
293
301
'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
294
302
'lzlv-70b': lzlv_70b,
295
303
'airoboros-70b': airoboros_70b,
296
'airoboros-l2-70b': airoboros_l2_70b,
297
304
'openchat_3.5': openchat_35,
298
305
'gemini': gemini,
299
306
'gemini-pro': gemini_pro,