From 0d1ae405cc24f3daafb783d758c019ef6731d3aa Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Thu, 26 Oct 2023 21:32:49 +0200
Subject: Add Llama2 Providers / Models

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

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index 7eee917a..45799602 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',
-- 
cgit v1.2.3