from transformers.models.auto.modeling_auto import _BaseAutoModelClass
from transformers.models.auto.configuration_auto import _LazyAutoMapping
from transformers.configuration_utils import PretrainedConfig
from transformers.models.auto import AutoModel

from synchformer_config import SynchformerConfig
from synchformer_model import SynchformerModel

# Register the model with the transformers library
AutoModel.register(SynchformerConfig, SynchformerModel)