Update unet/config.json

unet/config.json

@@ -4,6 +4,7 @@
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": null,
   "attention_head_dim": 8,
   "block_out_channels": [
     320,
@@ -51,6 +52,7 @@
   "time_embedding_dim": null,
   "time_embedding_type": "positional",
   "timestep_post_act": null,
+  "transformer_layers_per_block": 1,
   "up_block_types": [
     "UpBlock2D",
     "CrossAttnUpBlock2D",
@@ -59,4 +61,4 @@
   ],
   "upcast_attention": null,
   "use_linear_projection": false
-}
+}