provider: fix gpt2 name

This commit is contained in:
0xMRTT 2023-05-07 16:15:28 +02:00
parent af9bfc1d87
commit 72715e0bde
Signed by: 0xMRTT
GPG Key ID: 19C1449A774028BD
2 changed files with 2 additions and 2 deletions

View File

@ -2,7 +2,7 @@ from .huggingface import BaseHFProvider
class HuggingFaceGPT2LargeProvider(BaseHFProvider):
name = "GPT 2"
name = "GPT 2 Large"
slug = "hfgpt2large"
model = "gpt2-large"

View File

@ -2,7 +2,7 @@ from .huggingface import BaseHFProvider
class HuggingFaceGPT2XLProvider(BaseHFProvider):
name = "GPT 2"
name = "GPT 2 XL"
slug = "hfgpt2"
model = "gpt2-xl"