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())