dg845 commited on
Commit
7c63d54
1 Parent(s): 09e08ee

Upload 4 files

Browse files

Add new consistency models test U-Nets with mid block attention group norm fix

test_unet/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": 8,
 
7
  "block_out_channels": [
8
  32,
9
  64
@@ -15,6 +16,7 @@
15
  "AttnDownBlock2D"
16
  ],
17
  "downsample_padding": 1,
 
18
  "flip_sin_to_cos": true,
19
  "freq_shift": 0,
20
  "in_channels": 3,
@@ -27,10 +29,9 @@
27
  "resnet_time_scale_shift": "scale_shift",
28
  "sample_size": 32,
29
  "time_embedding_type": "positional",
30
- "upsample_type": "resnet",
31
- "downsample_type": "resnet",
32
  "up_block_types": [
33
  "AttnUpBlock2D",
34
  "ResnetUpsampleBlock2D"
35
- ]
 
36
  }
 
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
7
+ "attn_norm_num_groups": 32,
8
  "block_out_channels": [
9
  32,
10
  64
 
16
  "AttnDownBlock2D"
17
  ],
18
  "downsample_padding": 1,
19
+ "downsample_type": "resnet",
20
  "flip_sin_to_cos": true,
21
  "freq_shift": 0,
22
  "in_channels": 3,
 
29
  "resnet_time_scale_shift": "scale_shift",
30
  "sample_size": 32,
31
  "time_embedding_type": "positional",
 
 
32
  "up_block_types": [
33
  "AttnUpBlock2D",
34
  "ResnetUpsampleBlock2D"
35
+ ],
36
+ "upsample_type": "resnet"
37
  }
test_unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10fbd4bf346f20f687ec2a1654e6877d7fe8149e8bd621180353c06bea2d1f76
3
+ size 4633412
test_unet_class_cond/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": 8,
 
7
  "block_out_channels": [
8
  32,
9
  64
@@ -15,6 +16,7 @@
15
  "AttnDownBlock2D"
16
  ],
17
  "downsample_padding": 1,
 
18
  "flip_sin_to_cos": true,
19
  "freq_shift": 0,
20
  "in_channels": 3,
@@ -26,11 +28,10 @@
26
  "out_channels": 3,
27
  "resnet_time_scale_shift": "scale_shift",
28
  "sample_size": 32,
29
- "upsample_type": "resnet",
30
- "downsample_type": "resnet",
31
  "time_embedding_type": "positional",
32
  "up_block_types": [
33
  "AttnUpBlock2D",
34
  "ResnetUpsampleBlock2D"
35
- ]
 
36
  }
 
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
7
+ "attn_norm_num_groups": 32,
8
  "block_out_channels": [
9
  32,
10
  64
 
16
  "AttnDownBlock2D"
17
  ],
18
  "downsample_padding": 1,
19
+ "downsample_type": "resnet",
20
  "flip_sin_to_cos": true,
21
  "freq_shift": 0,
22
  "in_channels": 3,
 
28
  "out_channels": 3,
29
  "resnet_time_scale_shift": "scale_shift",
30
  "sample_size": 32,
 
 
31
  "time_embedding_type": "positional",
32
  "up_block_types": [
33
  "AttnUpBlock2D",
34
  "ResnetUpsampleBlock2D"
35
+ ],
36
+ "upsample_type": "resnet"
37
  }
test_unet_class_cond/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74c24e2a295ea2c52edf3c967ee330cfafa2fef53f82c943dde61a77655dc125
3
+ size 5145572