| # __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"] | |