返回提交历史
Modified
g4f/models.py
+25
-5
XFEstudio/gpt4free
Added CodeLlama and misc models from DeepInfra (#1506)
259f19e3
代码差异
1 个文件
+25
-5
@@ -120,10 +120,10 @@ llama2_70b = Model(
120
120
codellama_34b_instruct = Model(
121
121
name = "codellama/CodeLlama-34b-Instruct-hf",
122
122
base_provider = "huggingface",
123
best_provider = RetryProvider([HuggingChat, PerplexityLabs])
123
best_provider = RetryProvider([HuggingChat, PerplexityLabs, DeepInfra])
124
124
)
125
125
126
# Mistal
126
# Mistral
127
127
mixtral_8x7b = Model(
128
128
name = "mistralai/Mixtral-8x7B-Instruct-v0.1",
129
129
base_provider = "huggingface",
@@ -136,14 +136,31 @@ mistral_7b = Model(
136
136
best_provider = RetryProvider([DeepInfra, HuggingChat, PerplexityLabs])
137
137
)
138
138
139
# Dolphin
139
# Misc models
140
140
dolphin_mixtral_8x7b = Model(
141
141
name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b",
142
142
base_provider = "huggingface",
143
143
best_provider = DeepInfra
144
144
)
145
145
146
# OpenChat
146
lzlv_70b = Model(
147
name = "lizpreciatior/lzlv_70b_fp16_hf",
148
base_provider = "huggingface",
149
best_provider = DeepInfra
150
)
151
152
airoboros_70b = Model(
153
name = "deepinfra/airoboros-70b",
154
base_provider = "huggingface",
155
best_provider = DeepInfra
156
)
157
158
airoboros_l2_70b = Model(
159
name = "jondurbin/airoboros-l2-70b-gpt4-1.4.1",
160
base_provider = "huggingface",
161
best_provider = DeepInfra
162
)
163
147
164
openchat_35 = Model(
148
165
name = "openchat/openchat_3.5",
149
166
base_provider = "huggingface",
@@ -243,6 +260,9 @@ class ModelUtils:
243
260
'mixtral-8x7b': mixtral_8x7b,
244
261
'mistral-7b': mistral_7b,
245
262
'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
263
'lzlv-70b': lzlv_70b,
264
'airoboros-70b': airoboros_70b,
265
'airoboros-l2-70b': airoboros_l2_70b,
246
266
'openchat_3.5': openchat_35,
247
267
'gemini-pro': gemini_pro,
248
268
'bard': bard,
@@ -250,4 +270,4 @@ class ModelUtils:
250
270
'pi': pi
251
271
}
252
272
253
_all_models = list(ModelUtils.convert.keys())
273
_all_models = list(ModelUtils.convert.keys())