Update unet/config.json

unet/config.json

@@ -26,7 +26,7 @@
   "in_channels": 8,
   "lan_cross_attn": false,
   "layers_per_block": 2,
-  "mask_decoder_type": null,
+  "mask_decoder_type": "MaskDecoder",
   "mid_block_scale_factor": 1,
   "mid_block_type": "UNetMidBlock3DCrossAttn",
   "norm_eps": 1e-05,
@@ -37,7 +37,7 @@
   "ref_frame_type_2": null,
   "resnet_time_scale_shift": "default",
   "sample_size": 64,
-  "temp_attn": false,
+  "temp_attn": true,
   "up_block_types": [
     "UpBlock3D",
     "CrossAttnUpBlock3D",
@@ -46,4 +46,4 @@
   ],
   "upcast_attention": false,
   "use_linear_projection": false
-}
+}