Was asked to updated the sample_size to 64

#31
Files changed (1) hide show
  1. unet/config.json +1 -1
unet/config.json CHANGED
@@ -26,7 +26,7 @@
26
  "norm_eps": 1e-05,
27
  "norm_num_groups": 32,
28
  "out_channels": 4,
29
- "sample_size": 32,
30
  "up_block_types": [
31
  "UpBlock2D",
32
  "CrossAttnUpBlock2D",
 
26
  "norm_eps": 1e-05,
27
  "norm_num_groups": 32,
28
  "out_channels": 4,
29
+ "sample_size": 64,
30
  "up_block_types": [
31
  "UpBlock2D",
32
  "CrossAttnUpBlock2D",