provider(startcoder): add starcoder
This commit is contained in:
parent
e37fb02eeb
commit
5c3973c420
|
@ -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,
|
||||
}
|
||||
|
|
|
@ -22,6 +22,7 @@ providers_sources = [
|
|||
'openaigpt4.py',
|
||||
'openaigpt35turbo.py',
|
||||
'openaitextdavinci003.py',
|
||||
'starcoder.py',
|
||||
]
|
||||
|
||||
PY_INSTALLDIR.install_sources(providers_sources, subdir: providers_dir)
|
||||
|
|
7
src/provider/starcoder.py
Normal file
7
src/provider/starcoder.py
Normal file
|
@ -0,0 +1,7 @@
|
|||
from .huggingchatbase import BaseHuggingChatProvider
|
||||
|
||||
|
||||
class StarCoderProvider(BaseHuggingChatProvider):
|
||||
name = "Star Coder"
|
||||
slug = "starcoder"
|
||||
model = "bigcode/starcoder"
|
Loading…
Reference in New Issue
Block a user