patrickvonplaten commited on
Commit
75bf6b5
1 Parent(s): 20444fa

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +8 -6
unet/config.json CHANGED
@@ -14,9 +14,9 @@
14
  "class_embed_type": null,
15
  "down_block_types": [
16
  "ResnetDownsampleBlock2D",
17
- "AttnDownsampleBlock2D",
18
- "AttnDownsampleBlock2D",
19
- "AttnDownsampleBlock2D"
20
  ],
21
  "downsample_padding": 1,
22
  "flip_sin_to_cos": true,
@@ -31,10 +31,12 @@
31
  "resnet_time_scale_shift": "scale_shift",
32
  "sample_size": 64,
33
  "time_embedding_type": "positional",
 
 
34
  "up_block_types": [
35
- "AttnUpsampleBlock2D",
36
- "AttnUpsampleBlock2D",
37
- "AttnUpsampleBlock2D",
38
  "ResnetUpsampleBlock2D"
39
  ]
40
  }
 
14
  "class_embed_type": null,
15
  "down_block_types": [
16
  "ResnetDownsampleBlock2D",
17
+ "AttnDownBlock2D",
18
+ "AttnDownBlock2D",
19
+ "AttnDownBlock2D"
20
  ],
21
  "downsample_padding": 1,
22
  "flip_sin_to_cos": true,
 
31
  "resnet_time_scale_shift": "scale_shift",
32
  "sample_size": 64,
33
  "time_embedding_type": "positional",
34
+ "upsample_type": "resnet",
35
+ "downsample_type": "resnet",
36
  "up_block_types": [
37
+ "AttnUpBlock2D",
38
+ "AttnUpBlock2D",
39
+ "AttnUpBlock2D",
40
  "ResnetUpsampleBlock2D"
41
  ]
42
  }