summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authornullstreak <139914347+nullstreak@users.noreply.github.com>2023-12-13 13:15:27 +0100
committerGitHub <noreply@github.com>2023-12-13 13:15:27 +0100
commitc7b5e5eee67089e604f5ab11a7f58078e1639ec4 (patch)
tree0b2b9fac0ffeb90f8862653d679016f2fb893f41 /g4f/models.py
parent~ (diff)
downloadgpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar.gz
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar.bz2
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar.lz
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar.xz
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.tar.zst
gpt4free-c7b5e5eee67089e604f5ab11a7f58078e1639ec4.zip
Diffstat (limited to '')
-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())