Merge pull request #1488 from nullstreak/dolphin-mixtral

Add dolphin-2.6-mixtral-8x7b model
This commit is contained in:
Tekky 2024-01-21 10:46:25 +00:00 committed by GitHub
commit 29536bea9e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -130,6 +130,14 @@ mistral_7b = Model(
best_provider = RetryProvider([DeepInfra, HuggingChat]) best_provider = RetryProvider([DeepInfra, HuggingChat])
) )
# Dolphin
dolphin_mixtral_8x7b = Model(
name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b",
base_provider = "huggingface",
best_provider = DeepInfra
)
# OpenChat
openchat_35 = Model( openchat_35 = Model(
name = "openchat/openchat_3.5", name = "openchat/openchat_3.5",
base_provider = "huggingface", base_provider = "huggingface",
@ -337,6 +345,11 @@ class ModelUtils:
# Mistral # Mistral
'mixtral-8x7b': mixtral_8x7b, 'mixtral-8x7b': mixtral_8x7b,
'mistral-7b': mistral_7b, 'mistral-7b': mistral_7b,
# Dolphin
'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
# OpenChat
'openchat_3.5': openchat_35, 'openchat_3.5': openchat_35,
# Gemini Pro # Gemini Pro