summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authormadonchik123 <68397448+madonchik123@users.noreply.github.com>2023-12-01 23:11:52 +0100
committerGitHub <noreply@github.com>2023-12-01 23:11:52 +0100
commitb0276f6c9e2925c65a00b7189bad92feb25cefbd (patch)
tree1909ac2fd081e2e8ad5b5dfa8ae46bd6fd58c579 /g4f/models.py
parent~ (diff)
downloadgpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar.gz
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar.bz2
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar.lz
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar.xz
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.tar.zst
gpt4free-b0276f6c9e2925c65a00b7189bad92feb25cefbd.zip
Diffstat (limited to '')
-rw-r--r--g4f/models.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/g4f/models.py b/g4f/models.py
index 46eb49a0..e4b20ae5 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -27,6 +27,7 @@ from .Provider import (
Bing,
You,
H2o,
+ PI,
)
@dataclass(unsafe_hash=True)
@@ -260,6 +261,11 @@ llama70b_v2_chat = Model(
base_provider = 'replicate',
best_provider = Vercel)
+pi = Model(
+ name = 'pi',
+ base_provider = 'inflection',
+ best_provider=PI
+)
class ModelUtils:
convert: dict[str, Model] = {
@@ -315,6 +321,8 @@ class ModelUtils:
'oasst-sft-1-pythia-12b' : oasst_sft_1_pythia_12b,
'oasst-sft-4-pythia-12b-epoch-3.5' : oasst_sft_4_pythia_12b_epoch_35,
'command-light-nightly' : command_light_nightly,
+
+ 'pi': pi
}
_all_models = list(ModelUtils.convert.keys()) \ No newline at end of file