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

Update unet channels

Browse files
Files changed (1) hide show
  1. unet/config.json +1 -1
unet/config.json CHANGED
@@ -20,7 +20,7 @@
20
  "downsample_padding": 1,
21
  "flip_sin_to_cos": true,
22
  "freq_shift": 0,
23
- "in_channels": 4,
24
  "layers_per_block": 2,
25
  "mid_block_scale_factor": 1,
26
  "norm_eps": 1e-05,
 
20
  "downsample_padding": 1,
21
  "flip_sin_to_cos": true,
22
  "freq_shift": 0,
23
+ "in_channels": 9,
24
  "layers_per_block": 2,
25
  "mid_block_scale_factor": 1,
26
  "norm_eps": 1e-05,