provider(gpt2): add gpt2 xl and large

This commit is contained in:
0xMRTT 2023-05-07 16:08:29 +02:00
parent 4867f9c02d
commit cd5d73d64f
Signed by: 0xMRTT
GPG Key ID: 19C1449A774028BD
4 changed files with 28 additions and 0 deletions

View File

@ -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,

View 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
View 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

View File

@ -12,6 +12,8 @@ providers_sources = [
'hfgoogleflant5xxl.py',
'hfgoogleflanu12.py',
'hfgpt2.py',
'hfgpt2large.py',
'hfgpt2xl.py',
'hfopenassistantsft1pythia12b.py',
'huggingchat.py',
'huggingchatbase.py',