xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-12-24 03:23:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add Mistral models

Browse Source
This commit is contained in:
nullstreak 2023-12-13 13:15:27 +01:00 committed by GitHub
parent 90606ec7fc
commit c7b5e5eee6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
g4f/models.py
View File

@ -107,6 +107,17 @@ llama2_70b = Model(
base_provider = "huggingface", base_provider = "huggingface",
best_provider = RetryProvider([Llama2, DeepInfra, HuggingChat])) 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 # Bard
palm = Model( palm = Model(
name = 'palm', name = 'palm',
@ -292,6 +303,10 @@ class ModelUtils:
'llama2-13b': llama2_13b, 'llama2-13b': llama2_13b,
'llama2-70b': llama2_70b, 'llama2-70b': llama2_70b,
# Mistral
'mixtral-8x7b': mixtral_8x7b,
'mistral-7b': mistral_7b,
# Bard # Bard
'palm2' : palm, 'palm2' : palm,
'palm' : palm, 'palm' : palm,
@ -331,4 +346,4 @@ class ModelUtils:
'pi': pi 'pi': pi
} }
_all_models = list(ModelUtils.convert.keys()) _all_models = list(ModelUtils.convert.keys())