from .configuration_domsense import SystemOneConfig
from .modeling_domsense import SystemOneModelForDecision, DomSenseOutput
from .tokenization_domsense import DomSenseTokenizer

# 注册到 AutoXxx 工厂，便于：
#   AutoConfig      -> SystemOneConfig
#   AutoModel       -> SystemOneModelForDecision
#   AutoTokenizer   -> DomSenseTokenizer
from transformers import AutoConfig, AutoModel, AutoTokenizer

AutoConfig.register("domsense", SystemOneConfig)
AutoModel.register(SystemOneConfig, SystemOneModelForDecision)
AutoTokenizer.register(SystemOneConfig, DomSenseTokenizer)

__all__ = [
    "SystemOneConfig",
    "SystemOneModelForDecision",
    "DomSenseOutput",
    "DomSenseTokenizer",
]