provider(gpt2): add gpt2 xl and large
This commit is contained in:
parent
4867f9c02d
commit
cd5d73d64f
|
@ -11,6 +11,8 @@ from .hfopenassistantsft1pythia12b import HuggingFaceOpenAssistantSFT1PythiaProv
|
|||
from .hfgpt2 import HuggingFaceGPT2Provider
|
||||
from .hfdialogpt import HuggingFaceDialoGPTLargeProvider
|
||||
from .bard import BardProvider
|
||||
from .hfgpt2large import HuggingFaceGPT2LargeProvider
|
||||
from .hfgpt2xl import HuggingFaceGPT2XLProvider
|
||||
|
||||
PROVIDERS = {
|
||||
"alpacalora": AlpacaLoRAProvider,
|
||||
|
@ -21,6 +23,8 @@ PROVIDERS = {
|
|||
"hfgoogleflant5xxl": HuggingFaceGoogleFlanT5XXLProvider,
|
||||
"hfgoogleflanu12": HuggingFaceGoogleFlanU12Provider,
|
||||
"hfgpt2": HuggingFaceGPT2Provider,
|
||||
"hfgpt2large": HuggingFaceGPT2LargeProvider,
|
||||
"hfgpt2xl": HuggingFaceGPT2XLProvider,
|
||||
"hfopenassistantsft1pythia12b": HuggingFaceOpenAssistantSFT1PythiaProvider,
|
||||
"huggingchat": HuggingChatProvider,
|
||||
"openaigpt35turbo": OpenAIGPT35TurboProvider,
|
||||
|
|
11
src/provider/hfgpt2large.py
Normal file
11
src/provider/hfgpt2large.py
Normal file
|
@ -0,0 +1,11 @@
|
|||
from .huggingface import BaseHFProvider
|
||||
|
||||
|
||||
class HuggingFaceGPT2LargeProvider(BaseHFProvider):
|
||||
name = "GPT 2"
|
||||
slug = "hfgpt2large"
|
||||
model = "gpt2-large"
|
||||
|
||||
@property
|
||||
def require_api_key(self):
|
||||
return False
|
11
src/provider/hfgpt2xl.py
Normal file
11
src/provider/hfgpt2xl.py
Normal file
|
@ -0,0 +1,11 @@
|
|||
from .huggingface import BaseHFProvider
|
||||
|
||||
|
||||
class HuggingFaceGPT2XLProvider(BaseHFProvider):
|
||||
name = "GPT 2"
|
||||
slug = "hfgpt2"
|
||||
model = "gpt2-xl"
|
||||
|
||||
@property
|
||||
def require_api_key(self):
|
||||
return False
|
|
@ -12,6 +12,8 @@ providers_sources = [
|
|||
'hfgoogleflant5xxl.py',
|
||||
'hfgoogleflanu12.py',
|
||||
'hfgpt2.py',
|
||||
'hfgpt2large.py',
|
||||
'hfgpt2xl.py',
|
||||
'hfopenassistantsft1pythia12b.py',
|
||||
'huggingchat.py',
|
||||
'huggingchatbase.py',
|
||||
|
|
Loading…
Reference in New Issue
Block a user