msj19's picture
Add files using upload-large-folder tool
2efa03f verified
# -*- coding: utf-8 -*-
from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
from .configuration_emgla import emglaConfig
from .modeling_emgla import emglaForCausalLM, emglaModel
AutoConfig.register(emglaConfig.model_type, emglaConfig)
AutoModel.register(emglaConfig, emglaModel)
AutoModelForCausalLM.register(emglaConfig, emglaForCausalLM)
__all__ = ['emglaConfig', 'emglaForCausalLM', 'emglaModel']