from .modeling_synthesizer import SynthesizerConfig, Synthesizer
from transformers import AutoConfig, AutoModel

# Register custom config and model with Transformers
AutoConfig.register("synthesizer", SynthesizerConfig)
AutoModel.register(SynthesizerConfig, Synthesizer)