diff options
Diffstat (limited to 'g4f/models.py')
| -rw-r--r-- | g4f/models.py | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/g4f/models.py b/g4f/models.py index d7c29ccb..b34297f5 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -37,6 +37,10 @@ class Model: name: str base_provider: str best_provider: Union[type[BaseProvider], RetryProvider] = None + + @staticmethod + def __all__() -> list[str]: + return _all_models default = Model( name = "", @@ -231,6 +235,11 @@ llama7b_v2_chat = Model( base_provider = 'replicate', best_provider = Vercel) +llama70b_v2_chat = Model( + name = 'replicate/llama70b-v2-chat', + base_provider = 'replicate', + best_provider = Vercel) + class ModelUtils: convert: dict[str, Model] = { @@ -260,9 +269,9 @@ class ModelUtils: 'llama-13b' : llama_13b, # Vercel - 'claude-instant-v1' : claude_instant_v1, - 'claude-v1' : claude_v1, - 'claude-v2' : claude_v2, + #'claude-instant-v1' : claude_instant_v1, + #'claude-v1' : claude_v1, + #'claude-v2' : claude_v2, 'command-nightly' : command_nightly, 'gpt-neox-20b' : gpt_neox_20b, 'santacoder' : santacoder, @@ -274,6 +283,7 @@ class ModelUtils: 'text-curie-001' : text_curie_001, 'text-davinci-002' : text_davinci_002, 'text-davinci-003' : text_davinci_003, + 'llama70b-v2-chat' : llama70b_v2_chat, 'llama13b-v2-chat' : llama13b_v2_chat, 'llama7b-v2-chat' : llama7b_v2_chat, @@ -281,3 +291,5 @@ class ModelUtils: 'oasst-sft-4-pythia-12b-epoch-3.5' : oasst_sft_4_pythia_12b_epoch_35, 'command-light-nightly' : command_light_nightly, } + +_all_models = list(ModelUtils.convert.keys())
\ No newline at end of file |