provider(startcoder): add starcoder

This commit is contained in:
0xMRTT 2023-05-08 23:04:15 +02:00
parent e37fb02eeb
commit 5c3973c420
Signed by: 0xMRTT
GPG Key ID: 19C1449A774028BD
3 changed files with 10 additions and 0 deletions

View File

@ -13,6 +13,7 @@ from .hfdialogpt import HuggingFaceDialoGPTLargeProvider
from .bard import BardProvider
from .hfgpt2large import HuggingFaceGPT2LargeProvider
from .hfgpt2xl import HuggingFaceGPT2XLProvider
from .starcoder import StarCoderProvider
PROVIDERS = {
"alpacalora": AlpacaLoRAProvider,
@ -30,4 +31,5 @@ PROVIDERS = {
"openaigpt35turbo": OpenAIGPT35TurboProvider,
"openaigpt4": OpenAIGPT4Provider,
"openaitextdavinci003": OpenAITextDavinci003,
"starcoder": StarCoderProvider,
}

View File

@ -22,6 +22,7 @@ providers_sources = [
'openaigpt4.py',
'openaigpt35turbo.py',
'openaitextdavinci003.py',
'starcoder.py',
]
PY_INSTALLDIR.install_sources(providers_sources, subdir: providers_dir)

View File

@ -0,0 +1,7 @@
from .huggingchatbase import BaseHuggingChatProvider
class StarCoderProvider(BaseHuggingChatProvider):
name = "Star Coder"
slug = "starcoder"
model = "bigcode/starcoder"