diff options
Diffstat (limited to 'g4f/models.py')
| -rw-r--r-- | g4f/models.py | 28 |
1 files changed, 26 insertions, 2 deletions
diff --git a/g4f/models.py b/g4f/models.py index ae2ef54b..6fcfdd14 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -2,7 +2,7 @@ from __future__ import annotations from dataclasses import dataclass -from .Provider import RetryProvider, ProviderType +from .Provider import RetryProvider, ProviderType from .Provider import ( Chatgpt4Online, PerplexityLabs, @@ -15,6 +15,7 @@ from .Provider import ( ChatgptAi, DeepInfra, ChatBase, + GigaChat, Liaobots, FreeGpt, Llama2, @@ -95,6 +96,24 @@ gpt_4_turbo = Model( best_provider = Bing ) +gigachat = Model( + name = 'GigaChat:latest', + base_provider = 'gigachat', + best_provider = GigaChat +) + +gigachat_plus = Model( + name = 'GigaChat-Plus', + base_provider = 'gigachat', + best_provider = GigaChat +) + +gigachat_pro = Model( + name = 'GigaChat-Pro', + base_provider = 'gigachat', + best_provider = GigaChat +) + llama2_7b = Model( name = "meta-llama/Llama-2-7b-chat-hf", base_provider = 'meta', @@ -271,7 +290,12 @@ class ModelUtils: 'llama2-70b': llama2_70b, 'codellama-34b-instruct': codellama_34b_instruct, 'codellama-70b-instruct': codellama_70b_instruct, - + + # GigaChat + 'gigachat' : gigachat, + 'gigachat_plus': gigachat_plus, + 'gigachat_pro' : gigachat_pro, + 'mixtral-8x7b': mixtral_8x7b, 'mistral-7b': mistral_7b, 'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b, |