2024-02-25 16:52:27 +05:30
|
|
|
from .hfbasechat import BaseHFChatProvider, ProviderType
|
2023-08-03 05:56:13 +05:30
|
|
|
|
|
|
|
|
class DialoGPTProvider(BaseHFChatProvider):
|
|
|
|
|
name = "DialoGPT"
|
|
|
|
|
description = "A State-of-the-Art Large-scale Pretrained Response generation model"
|
|
|
|
|
provider = "microsoft/DialoGPT-large"
|
2024-02-25 16:52:27 +05:30
|
|
|
provider_type = ProviderType.CHAT
|
