patrickvonplaten dg845 commited on
Commit
fd221dd
1 Parent(s): dce8392

Upload config.json (#1)

Browse files

- Upload config.json (4e2aa3eeeaa8322a380eafae4a89ed1f1734df96)


Co-authored-by: Daniel Gu <dg845@users.noreply.huggingface.co>

diffusers_cd_imagenet64_l2/config.json CHANGED
@@ -1,9 +1,10 @@
1
  {
2
  "_class_name": "UNet2DModel",
3
- "_diffusers_version": "0.17.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 64,
 
7
  "block_out_channels": [
8
  192,
9
  384,
@@ -19,6 +20,7 @@
19
  "AttnDownBlock2D"
20
  ],
21
  "downsample_padding": 1,
 
22
  "flip_sin_to_cos": true,
23
  "freq_shift": 0,
24
  "in_channels": 3,
@@ -31,12 +33,11 @@
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
  }
 
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 64,
7
+ "attn_norm_num_groups": 32,
8
  "block_out_channels": [
9
  192,
10
  384,
 
20
  "AttnDownBlock2D"
21
  ],
22
  "downsample_padding": 1,
23
+ "downsample_type": "resnet",
24
  "flip_sin_to_cos": true,
25
  "freq_shift": 0,
26
  "in_channels": 3,
 
33
  "resnet_time_scale_shift": "scale_shift",
34
  "sample_size": 64,
35
  "time_embedding_type": "positional",
 
 
36
  "up_block_types": [
37
  "AttnUpBlock2D",
38
  "AttnUpBlock2D",
39
  "AttnUpBlock2D",
40
  "ResnetUpsampleBlock2D"
41
+ ],
42
+ "upsample_type": "resnet"
43
  }