ed83042 52bd532 ed83042 52bd532
1
2
3
4
5
6
7
8
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
