diff options
Diffstat (limited to 'g4f/models.py')
| -rw-r--r-- | g4f/models.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/g4f/models.py b/g4f/models.py index acd61846..e3da0363 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -29,6 +29,7 @@ from .Provider import ( Pi, Vercel, You, + Reka ) @@ -306,6 +307,12 @@ blackbox = Model( best_provider = Blackbox ) +reka_core = Model( + name = 'reka-core', + base_provider = 'Reka AI', + best_provider = Reka +) + class ModelUtils: """ Utility class for mapping string identifiers to Model instances. @@ -333,8 +340,12 @@ class ModelUtils: 'llama2-7b' : llama2_7b, 'llama2-13b': llama2_13b, 'llama2-70b': llama2_70b, + + 'llama3-8b' : llama3_8b_instruct, # alias + 'llama3-70b': llama3_70b_instruct, # alias 'llama3-8b-instruct' : llama3_8b_instruct, 'llama3-70b-instruct': llama3_70b_instruct, + 'codellama-34b-instruct': codellama_34b_instruct, 'codellama-70b-instruct': codellama_70b_instruct, @@ -359,6 +370,11 @@ class ModelUtils: 'claude-3-opus': claude_3_opus, 'claude-3-sonnet': claude_3_sonnet, + # reka core + 'reka-core': reka_core, + 'reka': reka_core, + 'Reka Core': reka_core, + # other 'blackbox': blackbox, 'command-r+': command_r_plus, |