summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/models.py')
-rw-r--r--g4f/models.py17
1 files changed, 16 insertions, 1 deletions
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())