| 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) |