From c7b5e5eee67089e604f5ab11a7f58078e1639ec4 Mon Sep 17 00:00:00 2001 From: nullstreak <139914347+nullstreak@users.noreply.github.com> Date: Wed, 13 Dec 2023 13:15:27 +0100 Subject: [PATCH] Add Mistral models --- g4f/models.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/g4f/models.py b/g4f/models.py index 810722e4..188cbf5c 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -107,6 +107,17 @@ llama2_70b = Model( base_provider = "huggingface", best_provider = RetryProvider([Llama2, DeepInfra, HuggingChat])) +# Mistal +mixtral_8x7b = Model( + name = "mistralai/Mixtral-8x7B-Instruct-v0.1", + base_provider = "huggingface", + best_provider = HuggingChat) + +mistral_7b = Model( + name = "mistralai/Mistral-7B-Instruct-v0.1", + base_provider = "huggingface", + best_provider = HuggingChat) + # Bard palm = Model( name = 'palm', @@ -292,6 +303,10 @@ class ModelUtils: 'llama2-13b': llama2_13b, 'llama2-70b': llama2_70b, + # Mistral + 'mixtral-8x7b': mixtral_8x7b, + 'mistral-7b': mistral_7b, + # Bard 'palm2' : palm, 'palm' : palm, @@ -331,4 +346,4 @@ class ModelUtils: 'pi': pi } -_all_models = list(ModelUtils.convert.keys()) \ No newline at end of file +_all_models = list(ModelUtils.convert.keys())