Upload AutoEncoder

config.json

@@ -3,7 +3,7 @@
     "AutoEncoder"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_autoencoder.AutoEncoderConfig",
+    "AutoConfig": "modeling_autoencoder.AutoEncoderConfig",
     "AutoModel": "modeling_autoencoder.AutoEncoder"
   },
   "compression_rate": 0.5,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c90ea0c944b1ef14b974e940fb6f688629657198fa2f2831d55527562a554807
+oid sha256:0843a9219f8207ea933b09d69c8fd229ed196f2427634d4a250e313c2e7cb790
 size 7348552

modeling_autoencoder.py

@@ -1,12 +1,35 @@
 
 from torch import nn
 
-from autoencoder_model.configuration_autoencoder import AutoEncoderConfig
+#from autoencoder_model.configuration_autoencoder import AutoEncoderConfig
+
+
 
 
 from transformers import PretrainedConfig, PreTrainedModel
 
+from transformers import PretrainedConfig
+
+class AutoEncoderConfig(PretrainedConfig):
+    model_type = "autoencoder"
 
+    def __init__(
+        self, 
+        input_dim=None, 
+        latent_dim=None, 
+        layer_types=None, 
+        dropout_rate=None, 
+        num_layers=None, 
+        compression_rate=None, 
+        **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.input_dim = input_dim
+        self.latent_dim = latent_dim
+        self.layer_types = layer_types
+        self.dropout_rate = dropout_rate
+        self.num_layers = num_layers
+        self.compression_rate = compression_rate
 
 def create_layers(model_section, layer_types, input_dim, latent_dim, num_layers, dropout_rate, compression_rate):