diff options
Diffstat (limited to 'g4f/models.py')
| -rw-r--r-- | g4f/models.py | 48 |
1 files changed, 31 insertions, 17 deletions
diff --git a/g4f/models.py b/g4f/models.py index 7c2d6822..cca9e850 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -3,25 +3,25 @@ from dataclasses import dataclass from .typing import Union from .Provider import BaseProvider, RetryProvider from .Provider import ( + AItianhuSpace, ChatgptLogin, - ChatgptAi, + PerplexityAi, + ChatgptDuo, + ChatgptAi, ChatBase, + AItianhu, + Wewordle, + Yqcloud, + Myshell, Vercel, DeepAi, + Aichat, Aivvm, - Bard, - H2o, GptGo, - Bing, - PerplexityAi, - Wewordle, - Yqcloud, - AItianhu, - AItianhuSpace, - Aichat, - Myshell, + Bard, Aibn, - ChatgptDuo, + Bing, + H2o, ) @dataclass(unsafe_hash=True) @@ -166,7 +166,17 @@ gpt_35_turbo_0613 = Model( gpt_4_0613 = Model( name = 'gpt-4-0613', base_provider = 'openai', - best_provider = Vercel) + best_provider = Aivvm) + +gpt_4_32k = Model( + name = 'gpt-4-32k', + base_provider = 'openai', + best_provider = Aivvm) + +gpt_4_32k_0613 = Model( + name = 'gpt-4-32k-0613', + base_provider = 'openai', + best_provider = Aivvm) text_ada_001 = Model( name = 'text-ada-001', @@ -206,13 +216,17 @@ llama7b_v2_chat = Model( class ModelUtils: convert: dict[str, Model] = { - # gpt-3.5 / gpt-4 + # gpt-3.5 'gpt-3.5-turbo' : gpt_35_turbo, 'gpt-3.5-turbo-16k' : gpt_35_turbo_16k, - 'gpt-4' : gpt_4, - 'gpt-4-0613' : gpt_4_0613, 'gpt-3.5-turbo-16k-0613' : gpt_35_turbo_16k_0613, + # gpt-4 + 'gpt-4' : gpt_4, + 'gpt-4-0613' : gpt_4_0613, + 'gpt-4-32k' : gpt_4_32k, + 'gpt-4-32k-0613' : gpt_4_32k_0613, + # Bard 'palm2' : palm, 'palm' : palm, @@ -247,4 +261,4 @@ 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, - }
\ No newline at end of file + } |