Upload folder using huggingface_hub

dvae/dvae_f.pth

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

dvae/model_config.json

@@ -0,0 +1,10 @@
+{
+    "enc_depth": 8,
+    "dec_depth": 8,
+    "dim": 256,
+    "output_dim": 48,
+    "latent_dim": 32,
+    "T": 16,
+    "num_traj": 512,
+    "n_layers": 1
+}

dvae_factors/dvae_f.json

@@ -0,0 +1,6 @@
+{
+    "f0_mean": 0.205227330327034,
+    "f0_std": 9.064417839050293,
+    "ft_mean": 0.95684415102005,
+    "ft_std": 6.397393703460693
+}

rf_model/rf_epoch_f.pth

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

rf_model/training_config.json

@@ -0,0 +1,31 @@
+{
+    "vae_config": {
+        "enc_depth": 8,
+        "dec_depth": 8,
+        "dim": 256,
+        "output_dim": 48,
+        "latent_dim": 32,
+        "T": 16,
+        "num_traj": 512,
+        "n_layers": 1
+    },
+    "rf_config": {
+        "width": 512,
+        "layers": 12,
+        "heads": 8,
+        "cond_drop_prob": 0.1,
+        "input_channels": 64,
+        "output_channels": 64,
+        "use_flash2": true
+    },
+    "training_args": {
+        "exp_name": "rf_model",
+        "vae_exp_dependency": "dvae",
+        "vae_epoch_dependency": "f",
+        "learning_rate": 0.0002,
+        "batch_size_per_gpu": 64,
+        "total_epochs": 2000,
+        "seed": 666,
+        "rescale_stats": true
+    }
+}