summaryrefslogtreecommitdiffstats
path: root/g4f
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-12-13 15:28:22 +0100
committerGitHub <noreply@github.com>2023-12-13 15:28:22 +0100
commit5d8e603095156303a016cc16e2811a8f2bc74f15 (patch)
treed9f14ffa57f9feacc5ea2d213cfb2de4ebbda5b0 /g4f
parentMerge pull request #1341 from GEOEGII555/patch-1 (diff)
parentAdd Mistral models (diff)
downloadgpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.gz
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.bz2
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.lz
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.xz
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.zst
gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.zip
Diffstat (limited to 'g4f')
-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())