__init__.py

from transformers import AutoConfig, AutoModel, CONFIG_MAPPING, MODEL_MAPPING
from .configuration_fireflies import FirefliesConfig
from .modeling_fireflies import FirefliesModel

CONFIG_MAPPING.register(FirefliesConfig.model_type, FirefliesConfig)
MODEL_MAPPING.register(FirefliesConfig, FirefliesModel)