Upload AutoEncoder1d

autoencoder.py

@@ -1,5 +1,5 @@
 from transformers import PreTrainedModel
-from package.audio_encoders_pytorch import AutoEncoder1d as AE1d, TanhBottleneck
+from audio_encoders_pytorch import AutoEncoder1d as AE1d, TanhBottleneck
 from .autoencoder_config import AutoEncoder1dConfig
 
 bottleneck = { 'tanh': TanhBottleneck }

config.json

@@ -1,6 +1,10 @@
 {
+  "architectures": [
+    "AutoEncoder1d"
+  ],
   "auto_map": {
-    "AutoConfig": "autoencoder_config.AutoEncoder1dConfig"
+    "AutoConfig": "autoencoder_config.AutoEncoder1dConfig",
+    "AutoModel": "autoencoder.AutoEncoder1d"
   },
   "bottleneck": "tanh",
   "channels": 32,
@@ -32,5 +36,6 @@
     8
   ],
   "patch_size": 4,
+  "torch_dtype": "float32",
   "transformers_version": "4.24.0"
 }