provider: fix gpt2 name
This commit is contained in:
parent
af9bfc1d87
commit
72715e0bde
|
@ -2,7 +2,7 @@ from .huggingface import BaseHFProvider
|
|||
|
||||
|
||||
class HuggingFaceGPT2LargeProvider(BaseHFProvider):
|
||||
name = "GPT 2"
|
||||
name = "GPT 2 Large"
|
||||
slug = "hfgpt2large"
|
||||
model = "gpt2-large"
|
||||
|
||||
|
|
|
@ -2,7 +2,7 @@ from .huggingface import BaseHFProvider
|
|||
|
||||
|
||||
class HuggingFaceGPT2XLProvider(BaseHFProvider):
|
||||
name = "GPT 2"
|
||||
name = "GPT 2 XL"
|
||||
slug = "hfgpt2"
|
||||
model = "gpt2-xl"
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user