__init__.py

from .configuration_apex import APEXConfig
from .modeling_apex import APEXModel

from transformers import AutoConfig, AutoModel
AutoConfig.register("apex", APEXConfig)
AutoModel.register(APEXConfig, APEXModel)