Archives/gpt4free
2
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-15 06:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update DeepInfra models

Browse Source
This commit is contained in:
nullstreak 2024-04-12 05:18:08 +02:00
parent df6ce3a531
commit 2e171fc290
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
g4f/models.py
View File

@ -127,7 +127,7 @@ llama2_70b = Model(
codellama_34b_instruct = Model( codellama_34b_instruct = Model(
name = "codellama/CodeLlama-34b-Instruct-hf", name = "codellama/CodeLlama-34b-Instruct-hf",
base_provider = "meta", base_provider = "meta",
best_provider = RetryProvider([HuggingChat, DeepInfra]) best_provider = HuggingChat
) )
codellama_70b_instruct = Model( codellama_70b_instruct = Model(
@ -146,9 +146,21 @@ mixtral_8x7b = Model(
mistral_7b = Model( mistral_7b = Model(
name = "mistralai/Mistral-7B-Instruct-v0.1", name = "mistralai/Mistral-7B-Instruct-v0.1",
base_provider = "huggingface", base_provider = "huggingface",
best_provider = RetryProvider([DeepInfra, HuggingChat, HuggingFace, PerplexityLabs]) best_provider = RetryProvider([HuggingChat, HuggingFace, PerplexityLabs])
) )
mistral_7b_v02 = Model(
name = "mistralai/Mistral-7B-Instruct-v0.2",
base_provider = "huggingface",
best_provider = DeepInfra
)
# mixtral_8x22b = Model(
# name = "mistralai/Mixtral-8x22B-v0.1",
# base_provider = "huggingface",
# best_provider = DeepInfra
# )
# Misc models # Misc models
dolphin_mixtral_8x7b = Model( dolphin_mixtral_8x7b = Model(
name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b", name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b",
@ -168,12 +180,6 @@ airoboros_70b = Model(
best_provider = DeepInfra best_provider = DeepInfra
) )
airoboros_l2_70b = Model(
name = "jondurbin/airoboros-l2-70b-gpt4-1.4.1",
base_provider = "huggingface",
best_provider = DeepInfra
)
openchat_35 = Model( openchat_35 = Model(
name = "openchat/openchat_3.5", name = "openchat/openchat_3.5",
base_provider = "huggingface", base_provider = "huggingface",
@ -290,10 +296,11 @@ class ModelUtils:
'mixtral-8x7b': mixtral_8x7b, 'mixtral-8x7b': mixtral_8x7b,
'mistral-7b': mistral_7b, 'mistral-7b': mistral_7b,
'mistral-7b-v02': mistral_7b_v02,
# 'mixtral-8x22b': mixtral_8x22b,
'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b, 'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
'lzlv-70b': lzlv_70b, 'lzlv-70b': lzlv_70b,
'airoboros-70b': airoboros_70b, 'airoboros-70b': airoboros_70b,
'airoboros-l2-70b': airoboros_l2_70b,
'openchat_3.5': openchat_35, 'openchat_3.5': openchat_35,
'gemini': gemini, 'gemini': gemini,
'gemini-pro': gemini_pro, 'gemini-pro': gemini_pro,