Uploading VAE in neuro-symbolic-ai/eb-optimus

README.md

@@ -0,0 +1,13 @@
+---
+language: en
+tags:
+- pythae
+license: apache-2.0
+---
+
+### Downloading this model from the Hub
+This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
+```python
+>>> from pythae.models import AutoModel
+>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
+```

decoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d7d518482a49e5d5298750109d329b7ebd95752c3fd6f53fdba191be6706e9f
+size 517476137

encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:951c44b25d8612c7c1d6cbe8c8b4020145eb0c8f6ee6df90a0ea7f4e6a850c8c
+size 435344298

environment.json

@@ -0,0 +1 @@
+{"name": "EnvironmentConfig", "python_version": "3.11"}

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b741bf12b68dfd1453cb6fd4d662179b03afc3b3c2b5e13624d1c636f71eea40
+size 936993706

model_config.json

@@ -0,0 +1 @@
+{"name": "VAEConfig", "input_dim": [32, 50257], "latent_dim": 32, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse"}