liuhaotian commited on
Commit
7cb9a5e
1 Parent(s): 12ae27b

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +1 -0
unet/config.json CHANGED
@@ -18,6 +18,7 @@
18
  "DownBlock2D"
19
  ],
20
  "downsample_padding": 1,
 
21
  "flip_sin_to_cos": true,
22
  "freq_shift": 0,
23
  "in_channels": 9,
 
18
  "DownBlock2D"
19
  ],
20
  "downsample_padding": 1,
21
+ "use_gated_attention": true,
22
  "flip_sin_to_cos": true,
23
  "freq_shift": 0,
24
  "in_channels": 9,