File size: 298 Bytes
127b6bf | 1 2 3 4 5 6 7 8 | from .configuration_sproto import SprotoConfig
from .modeling_sproto import SprotoModel, SprotoOutput
from transformers import AutoConfig, AutoModel
AutoConfig.register("sproto", SprotoConfig)
AutoModel.register(SprotoConfig, SprotoModel)
__all__ = ["SprotoConfig", "SprotoModel", "SprotoOutput"] |