sproto / __init__.py
RamezCh's picture
Upload folder using huggingface_hub
127b6bf verified
Raw
History Blame Contribute Delete
298 Bytes
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"]