From c578385cc4f8c2f9608f500d0506645210266973 Mon Sep 17 00:00:00 2001 From: nullstreak <139914347+nullstreak@users.noreply.github.com> Date: Sun, 21 Jan 2024 06:20:45 +0100 Subject: [PATCH] Add dolphin-2.6-mixtral-8x7b --- g4f/models.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/g4f/models.py b/g4f/models.py index dd6e0a2c..40a72d77 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -130,6 +130,14 @@ mistral_7b = Model( 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( name = "openchat/openchat_3.5", base_provider = "huggingface", @@ -337,6 +345,11 @@ class ModelUtils: # Mistral 'mixtral-8x7b': mixtral_8x7b, 'mistral-7b': mistral_7b, + + # Dolphin + 'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b, + + # OpenChat 'openchat_3.5': openchat_35, # Gemini Pro