provider: fix slug
This commit is contained in:
parent
75f4578bce
commit
ad287c34c2
|
@ -8,7 +8,7 @@ from gi.repository import Gtk, Adw, GLib
|
|||
|
||||
class HuggingFaceDialoGPTLargeProvider(BaseHFProvider):
|
||||
name = "DialoGPT"
|
||||
slug = "dialogpt"
|
||||
slug = "hfdialogpt"
|
||||
model = "microsoft/DialoGPT-large"
|
||||
|
||||
@property
|
||||
|
|
|
@ -3,7 +3,7 @@ from .huggingface import BaseHFProvider
|
|||
|
||||
class HuggingFaceGPT2Provider(BaseHFProvider):
|
||||
name = "GPT 2"
|
||||
slug = "gpt2"
|
||||
slug = "hfgpt2"
|
||||
model = "gpt2"
|
||||
|
||||
@property
|
||||
|
|
Loading…
Reference in New Issue
Block a user