Upload AutoEncoder

config.json

@@ -10,10 +10,10 @@
   "compression_rate": 0.5,
   "dropout_rate": 0.1,
   "input_dim": 128,
-  "latent_dim": 32,
+  "latent_dim": 64,
   "layer_types": "linear",
   "model_type": "autoencoder",
-  "num_layers": 2,
+  "num_layers": 3,
   "torch_dtype": "float32",
   "transformers_version": "4.35.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:81f3c0326c30137a113ac0434b67dd54eb8505c7d4f801a1597629712ae7dd71
-size 83728
+oid sha256:11d5ab1ed136a95d5d14f711cd1d8e976dcc0480184b7dc45540af8404dafc5a
+size 133840

modeling_autoencoder.py

@@ -15,6 +15,11 @@ from transformers import PretrainedConfig, PreTrainedModel, AutoConfig, AutoMode
 
 # autoencoder.push_to_hub("autoencoder")
 
+# from transformers import AutoConfig, AutoModel
+ 
+# config = AutoConfig.from_pretrained("amaye15/autoencoder", trust_remote_code = True)
+# autoencoder = AutoModel.from_config(config, trust_remote_code = True)
+
 
 class AutoEncoderConfig(PretrainedConfig):
     """