diff options
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 |