| 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", | |
| ] | |