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