from transformers import AutoConfig, AutoModel from .configuration_emuru import EmuruConfig from .modeling_emuru import Emuru AutoConfig.register(EmuruConfig.model_type, EmuruConfig) AutoModel.register(EmuruConfig, Emuru)