patrickvonplaten commited on
Commit
6508280
1 Parent(s): 4668ddc

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +8 -6
unet/config.json CHANGED
@@ -18,9 +18,9 @@
18
  "ResnetDownsampleBlock2D",
19
  "ResnetDownsampleBlock2D",
20
  "ResnetDownsampleBlock2D",
21
- "AttnDownsampleBlock2D",
22
- "AttnDownsampleBlock2D",
23
- "AttnDownsampleBlock2D"
24
  ],
25
  "downsample_padding": 1,
26
  "flip_sin_to_cos": true,
@@ -35,10 +35,12 @@
35
  "resnet_time_scale_shift": "default",
36
  "sample_size": 256,
37
  "time_embedding_type": "positional",
 
 
38
  "up_block_types": [
39
- "AttnUpsampleBlock2D",
40
- "AttnUpsampleBlock2D",
41
- "AttnUpsampleBlock2D",
42
  "ResnetUpsampleBlock2D",
43
  "ResnetUpsampleBlock2D",
44
  "ResnetUpsampleBlock2D"
 
18
  "ResnetDownsampleBlock2D",
19
  "ResnetDownsampleBlock2D",
20
  "ResnetDownsampleBlock2D",
21
+ "AttnDownBlock2D",
22
+ "AttnDownBlock2D",
23
+ "AttnDownBlock2D"
24
  ],
25
  "downsample_padding": 1,
26
  "flip_sin_to_cos": true,
 
35
  "resnet_time_scale_shift": "default",
36
  "sample_size": 256,
37
  "time_embedding_type": "positional",
38
+ "upsample_type": "resnet",
39
+ "downsample_type": "resnet",
40
  "up_block_types": [
41
+ "AttnUpBlock2D",
42
+ "AttnUpBlock2D",
43
+ "AttnUpBlock2D",
44
  "ResnetUpsampleBlock2D",
45
  "ResnetUpsampleBlock2D",
46
  "ResnetUpsampleBlock2D"