|
|
|
@ -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())
|
|
|
|
|