Add Llama2 Providers / Models

This commit is contained in:
Heiner Lohaus 2023-10-26 21:32:49 +02:00
parent 92a31d8281
commit 0d1ae405cc
4 changed files with 91 additions and 9 deletions

View file

@ -6,12 +6,14 @@ from .Provider import (
GptForLove,
ChatgptAi,
GptChatly,
DeepInfra,
ChatgptX,
ChatBase,
GeekGpt,
FakeGpt,
FreeGpt,
NoowAi,
Llama2,
Vercel,
Aichat,
GPTalk,
@ -74,6 +76,21 @@ gpt_4 = Model(
])
)
llama2_7b = Model(
name = "meta-llama/Llama-2-7b-chat-hf",
base_provider = 'huggingface',
best_provider = RetryProvider([Llama2, DeepInfra]))
llama2_13b = Model(
name ="meta-llama/Llama-2-13b-chat-hf",
base_provider = 'huggingface',
best_provider = RetryProvider([Llama2, DeepInfra]))
llama2_70b = Model(
name = "meta-llama/Llama-2-70b-chat-hf",
base_provider = "huggingface",
best_provider = RetryProvider([Llama2, DeepInfra]))
# Bard
palm = Model(
name = 'palm',