Merge pull request #1343 from nullstreak/main

Add Mistral models
pull/1344/head
Tekky 10 months ago committed by GitHub
commit 5d8e603095
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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())
_all_models = list(ModelUtils.convert.keys())

Loading…
Cancel
Save