From 455e5dc9ea14480f029e1c08f5ff9298e0f9e085 Mon Sep 17 00:00:00 2001
From: nullstreak <139914347+nullstreak@users.noreply.github.com>
Date: Sun, 4 Feb 2024 19:50:32 +0100
Subject: Added CodeLlama 70b

---
 g4f/models.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'g4f')

diff --git a/g4f/models.py b/g4f/models.py
index e58ccef2..dd8e175d 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -123,6 +123,12 @@ codellama_34b_instruct = Model(
     best_provider = RetryProvider([HuggingChat, PerplexityLabs, DeepInfra])
 )
 
+codellama_70b_instruct = Model(
+    name          = "codellama/CodeLlama-70b-Instruct-hf",
+    base_provider = "huggingface",
+    best_provider = DeepInfra
+)
+
 # Mistral
 mixtral_8x7b = Model(
     name          = "mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -256,6 +262,7 @@ class ModelUtils:
         'llama2-13b': llama2_13b,
         'llama2-70b': llama2_70b,
         'codellama-34b-instruct': codellama_34b_instruct,
+        'codellama-70b-instruct': codellama_70b_instruct,
         
         'mixtral-8x7b': mixtral_8x7b,
         'mistral-7b': mistral_7b,
@@ -270,4 +277,4 @@ class ModelUtils:
         'pi': pi
     }
 
-_all_models = list(ModelUtils.convert.keys())
\ No newline at end of file
+_all_models = list(ModelUtils.convert.keys())
-- 
cgit v1.2.3