Indivec / __init__.py
Sailesh Panda
Registering model Architecture.
aeb7137
from .configuration_hinvec import HinvecConfig
from .modeling_hinvec import HinvecModel
from .tokenization_hinvec import HinvecTokenizer
from transformers import AutoConfig, AutoModel
__all__ = [
"HinvecConfig",
"HinvecModel",
"HinvecTokenizer",
]
AutoConfig.register("hinvec", HinvecConfig)
AutoModel.register(HinvecConfig, HinvecModel)