Uploading VAE in neuro-symbolic-ai/eb-langvae-stella_en_1.5B_v5-Mistral-7B-v0.3-l128

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.pt

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

decoder_cfg.json

@@ -0,0 +1 @@
+{"model_path": "mistralai/Mistral-7B-v0.3", "latent_size": 128, "max_len": 32, "conditional": false, "device_map": null}

encoder.pt

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

encoder_cfg.json

@@ -0,0 +1 @@
+{"model_path": "NovaSearch/stella_en_1.5B_v5", "latent_size": 128, "automodel_preset": {"cls": "AutoModel", "pooling_method": "mean", "normalize": true}, "caching": true}

environment.json

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

model_config.json

@@ -0,0 +1 @@
+{"name": "VAEConfig", "input_dim": null, "latent_dim": 128, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse"}