|
from .hfbasechat import BaseHFChatProvider, ProviderType
|
|
|
|
class DialoGPTProvider(BaseHFChatProvider):
|
|
name = "DialoGPT"
|
|
description = "A State-of-the-Art Large-scale Pretrained Response generation model"
|
|
provider = "microsoft/DialoGPT-large"
|
|
provider_type = ProviderType.CHAT
|