Fix: change of modeling and config to autoencoder type

config.json

@@ -1,9 +1,4 @@
 {
-  "model_type": "conv_autoencoder",
-  "latent_dim": 256,
-
-  "auto_map": {
-    "AutoModel": "modeling_autoencoder.AutoencoderModel",
-    "AutoConfig": "modeling_autoencoder.AutoencoderConfig"
-  }
+  "model_type": "autoencoder",
+  "latent_dim": 256
 }

configuration_conv_autoencoder.py → configuration_autoencoder.py

@@ -1,9 +1,8 @@
 from transformers import PretrainedConfig
 
-class ConvAutoencoderConfig(PretrainedConfig):
-    model_type = "conv_autoencoder"
+class AutoencoderConfig(PretrainedConfig):
+    model_type = "autoencoder"
 
     def __init__(self, latent_dim=256, **kwargs):
         super().__init__(**kwargs)
-        self.latent_dim = latent_dim
-        
+        self.latent_dim = latent_dim