backup

src/models/__init__.py

@@ -1,8 +1,10 @@
 from .Model import Model
+from .autoencoder import AutoEncoder
 from .cnn import CNNPredictor
 from .transformer import ByteTransformer
 
 model_called: dict[str, type[Model]] = {
     'cnn': CNNPredictor,
-    'transformer': ByteTransformer
+    'transformer': ByteTransformer,
+    'autoencoder': AutoEncoder
 }