diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-03-15 13:44:15 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-15 13:44:15 +0100 |
| commit | 7df7971d202af6e88010f17e83740d65386699ca (patch) | |
| tree | da8ffc82319214511fb34e404b5b4851c1fef02d /g4f/models.py | |
| parent | Error checking for Vercel (#1704) (diff) | |
| parent | GigaChat models add (diff) | |
| download | gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar.gz gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar.bz2 gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar.lz gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar.xz gpt4free-7df7971d202af6e88010f17e83740d65386699ca.tar.zst gpt4free-7df7971d202af6e88010f17e83740d65386699ca.zip | |
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, |