inital commit

__init__.py

@@ -0,0 +1,2 @@
+from .configuration_vae import VAEConfig
+from .modeling_vae import VAEModel

configuration_vae.py

@@ -0,0 +1,21 @@
+from transformers import PretrainedConfig
+from typing import Optional, Literal
+
+class VAEConfig(PretrainedConfig):
+    model_type = "vae"
+    
+    def __init__(
+        self,
+        data_dim=2,
+        latent_dim=2,
+        hidden_dim=96,
+        encoder_layers=2,
+        data_type: Optional[Literal['binary', 'continuous', 'auto']] = 'auto',
+        **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.data_dim = data_dim
+        self.latent_dim = latent_dim
+        self.hidden_dim = hidden_dim
+        self.encoder_layers = encoder_layers
+        self.data_type = data_type