Update config.yaml

config.yaml

@@ -34,6 +34,7 @@ model:
         out_channels: 4
         model_channels: 4
         channel_mult: [ 1, 2 ]
+        attention_resolutions: [2, 1]
         num_heads: 8
         num_res_blocks: 1
         use_spatial_transformer: True