Add llama2 model mapping

8 months ago · ba42301393
parent 0d1ae405cc
commit ba42301393
1 changed files with 5 additions and 0 deletions
--- 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,