from transformers import PretrainedConfig

class AutoencoderConfig(PretrainedConfig):
    model_type = "autoencoder"

    def __init__(self, latent_dim=256, **kwargs):
        super().__init__(**kwargs)
        self.latent_dim = latent_dim