summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/models.py')
-rw-r--r--g4f/models.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/g4f/models.py b/g4f/models.py
index 7eee917a..5e4d9b40 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -6,12 +6,14 @@ from .Provider import (
GptForLove,
ChatgptAi,
GptChatly,
+ DeepInfra,
ChatgptX,
ChatBase,
GeekGpt,
FakeGpt,
FreeGpt,
NoowAi,
+ Llama2,
Vercel,
Aichat,
GPTalk,
@@ -74,6 +76,21 @@ gpt_4 = Model(
])
)
+llama2_7b = Model(
+ name = "meta-llama/Llama-2-7b-chat-hf",
+ base_provider = 'huggingface',
+ best_provider = RetryProvider([Llama2, DeepInfra]))
+
+llama2_13b = Model(
+ name ="meta-llama/Llama-2-13b-chat-hf",
+ base_provider = 'huggingface',
+ best_provider = RetryProvider([Llama2, DeepInfra]))
+
+llama2_70b = Model(
+ name = "meta-llama/Llama-2-70b-chat-hf",
+ base_provider = "huggingface",
+ best_provider = RetryProvider([Llama2, DeepInfra]))
+
# Bard
palm = Model(
name = 'palm',
@@ -246,6 +263,11 @@ class ModelUtils:
'gpt-4-0613' : gpt_4_0613,
'gpt-4-32k' : gpt_4_32k,
'gpt-4-32k-0613' : gpt_4_32k_0613,
+
+ # Llama 2
+ 'llama2-7b' : llama2_7b,
+ 'llama2-13b': llama2_13b,
+ 'llama2-70b': llama2_70b,
# Bard
'palm2' : palm,