# __init__.py
from .modeling_custom import CustomConfig, CustomModel
from transformers import AutoConfig, AutoModel, CONFIG_MAPPING, MODEL_MAPPING

CONFIG_MAPPING.update({"custom_model": CustomConfig})
MODEL_MAPPING.update({"custom_model": CustomModel})

__all__ = ["CustomConfig", "CustomModel"]