Upload config.json

vae/config.json

@@ -1,24 +1,35 @@
-model:
-  base_learning_rate: 1.5e-7
-  target: ldm.models.autoencoder.AutoencoderKL
-  params:
-    monitor: "val/rec_loss"
-    embed_dim: 4
-    lossconfig:
-      target: ldm.modules.losses.LPIPSWithDiscriminator
-      params:
-        disc_start: 50001
-        kl_weight: 0.000001
-        disc_weight: 0.5
-
-    ddconfig:
-      double_z: True
-      z_channels: 4
-      resolution: 256
-      in_channels: 3
-      out_ch: 3
-      ch: 128
-      ch_mult: [ 1,2,4,4 ]  # num_down = len(ch_mult)-1
-      num_res_blocks: 2
-      attn_resolutions: [ ]
-      dropout: 0.0
+{
+  "model": {
+    "base_learning_rate": 1.5e-7,
+    "target": "ldm.models.autoencoder.AutoencoderKL",
+    "params": {
+      "monitor": "val/rec_loss",
+      "embed_dim": 4,
+      "lossconfig": {
+        "target": "ldm.modules.losses.LPIPSWithDiscriminator",
+        "params": {
+          "disc_start": 50001,
+          "kl_weight": 0.000001,
+          "disc_weight": 0.5
+        }
+      },
+      "ddconfig": {
+        "double_z": true,
+        "z_channels": 4,
+        "resolution": 256,
+        "in_channels": 3,
+        "out_ch": 3,
+        "ch": 128,
+        "ch_mult": [
+          1,
+          2,
+          4,
+          4
+        ],
+        "num_res_blocks": 2,
+        "attn_resolutions": [],
+        "dropout": 0
+      }
+    }
+  }
+}