summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authorvlad <master.redstone@yandex.ru>2024-03-13 23:29:49 +0100
committervlad <master.redstone@yandex.ru>2024-03-13 23:29:49 +0100
commitf07173de4fe64bdccfdb9adff34ae0abbfe70a4d (patch)
tree52d69fc6f9a656eb6d48c5fecda7d4b4c2cd2c28 /g4f/models.py
parentMerge pull request #1684 from hlohaus/retry (diff)
downloadgpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar.gz
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar.bz2
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar.lz
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar.xz
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.tar.zst
gpt4free-f07173de4fe64bdccfdb9adff34ae0abbfe70a4d.zip
Diffstat (limited to '')
-rw-r--r--g4f/models.py28
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,