Upload AutoEncoder

config.json

@@ -7,15 +7,18 @@
     "AutoModel": "modeling_autoencoder.AutoEncoder"
   },
   "bidirectional": false,
+  "bos_token_id": 1,
   "compression_rate": 0.5,
   "dropout_rate": 0.1,
   "embed": false,
+  "eos_token_id": 2,
   "input_dim": 128,
   "latent_dim": 64,
   "layer_types": "linear",
   "max_position": false,
   "model_type": "autoencoder",
   "num_layers": 3,
+  "pad_token_id": 0,
   "torch_dtype": "float32",
   "transformers_version": "4.35.2",
   "vocab_size": false

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:416ef0961369dfbc2337037fa0a017862a9d07f7fba1bce8e0a1fd09af91b433
+oid sha256:359d4bc07fabf4f4a00c60c8a28a185ff7813a185a6f620d42884833c0546e0e
 size 133840

modeling_autoencoder.py

@@ -116,6 +116,9 @@ class AutoEncoderConfig(PretrainedConfig):
         self.embed = embed
         self.vocab_size = vocab_size
         self.max_position = max_position
+        self.pad_token_id = pad_token_id
+        self.bos_token_id = bos_token_id
+        self.eos_token_id = eos_token_id
 
         if self.embed:
             if not self.vocab_size and isinstance(self.vocab_size, int):