summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/models.py')
-rw-r--r--g4f/models.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/g4f/models.py b/g4f/models.py
index 45799602..5e4d9b40 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -263,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,