From ba42301393103b99166ea1325b71204141077e0f Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Thu, 26 Oct 2023 21:42:30 +0200
Subject: Add llama2 model mapping

---
 g4f/models.py | 5 +++++
 1 file changed, 5 insertions(+)

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,
-- 
cgit v1.2.3