Push model using huggingface_hub.

config.json

@@ -7,5 +7,39 @@
   "offset_noise_strength": 0.0,
   "sampling_timesteps": 20,
   "schedule_fn_kwargs": {},
-  "timesteps": 50
+  "stldm_param": {
+    "base_ch": 64,
+    "chs_mult": [
+      1,
+      2,
+      4,
+      8
+    ],
+    "dim_head": 32,
+    "heads": 4,
+    "in_ch": 32,
+    "num_groups": 8,
+    "patch_size": 16
+  },
+  "timesteps": 50,
+  "vp_param": {
+    "N_S": 4,
+    "N_T": 8,
+    "groups": 8,
+    "hid_S": 32,
+    "hid_T": 512,
+    "last_activation": "sigmoid",
+    "shape_in": [
+      5,
+      1,
+      128,
+      128
+    ],
+    "shape_out": [
+      20,
+      1,
+      128,
+      128
+    ]
+  }
 }