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)