diff options
| author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-12-13 15:28:22 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-13 15:28:22 +0100 |
| commit | 5d8e603095156303a016cc16e2811a8f2bc74f15 (patch) | |
| tree | d9f14ffa57f9feacc5ea2d213cfb2de4ebbda5b0 /g4f | |
| parent | Merge pull request #1341 from GEOEGII555/patch-1 (diff) | |
| parent | Add Mistral models (diff) | |
| download | gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.gz gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.bz2 gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.lz gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.xz gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.tar.zst gpt4free-5d8e603095156303a016cc16e2811a8f2bc74f15.zip | |
Diffstat (limited to 'g4f')
| -rw-r--r-- | g4f/models.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/g4f/models.py b/g4f/models.py index 810722e4..188cbf5c 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -107,6 +107,17 @@ llama2_70b = Model( base_provider = "huggingface", best_provider = RetryProvider([Llama2, DeepInfra, HuggingChat])) +# Mistal +mixtral_8x7b = Model( + name = "mistralai/Mixtral-8x7B-Instruct-v0.1", + base_provider = "huggingface", + best_provider = HuggingChat) + +mistral_7b = Model( + name = "mistralai/Mistral-7B-Instruct-v0.1", + base_provider = "huggingface", + best_provider = HuggingChat) + # Bard palm = Model( name = 'palm', @@ -292,6 +303,10 @@ class ModelUtils: 'llama2-13b': llama2_13b, 'llama2-70b': llama2_70b, + # Mistral + 'mixtral-8x7b': mixtral_8x7b, + 'mistral-7b': mistral_7b, + # Bard 'palm2' : palm, 'palm' : palm, @@ -331,4 +346,4 @@ class ModelUtils: 'pi': pi } -_all_models = list(ModelUtils.convert.keys())
\ No newline at end of file +_all_models = list(ModelUtils.convert.keys()) |