diff options
| author | madonchik123 <68397448+madonchik123@users.noreply.github.com> | 2023-12-01 23:11:52 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-01 23:11:52 +0100 |
| commit | b0276f6c9e2925c65a00b7189bad92feb25cefbd (patch) | |
| tree | 1909ac2fd081e2e8ad5b5dfa8ae46bd6fd58c579 /g4f/models.py | |
| parent | ~ (diff) | |
| download | gpt4free-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 'g4f/models.py')
| -rw-r--r-- | g4f/models.py | 8 |
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 |