Upload 4 files

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

test_unet/config.json

@@ -1,9 +1,10 @@
 {
   "_class_name": "UNet2DModel",
-  "_diffusers_version": "0.17.0.dev0",
+  "_diffusers_version": "0.21.0.dev0",
   "act_fn": "silu",
   "add_attention": true,
   "attention_head_dim": 8,
+  "attn_norm_num_groups": 32,
   "block_out_channels": [
     32,
     64
@@ -15,6 +16,7 @@
     "AttnDownBlock2D"
   ],
   "downsample_padding": 1,
+  "downsample_type": "resnet",
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 3,
@@ -27,10 +29,9 @@
   "resnet_time_scale_shift": "scale_shift",
   "sample_size": 32,
   "time_embedding_type": "positional",
-  "upsample_type": "resnet",
-  "downsample_type": "resnet",
   "up_block_types": [
     "AttnUpBlock2D",
     "ResnetUpsampleBlock2D"
-  ]
+  ],
+  "upsample_type": "resnet"
 }

test_unet/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:10fbd4bf346f20f687ec2a1654e6877d7fe8149e8bd621180353c06bea2d1f76
+size 4633412

test_unet_class_cond/config.json

@@ -1,9 +1,10 @@
 {
   "_class_name": "UNet2DModel",
-  "_diffusers_version": "0.17.0.dev0",
+  "_diffusers_version": "0.21.0.dev0",
   "act_fn": "silu",
   "add_attention": true,
   "attention_head_dim": 8,
+  "attn_norm_num_groups": 32,
   "block_out_channels": [
     32,
     64
@@ -15,6 +16,7 @@
     "AttnDownBlock2D"
   ],
   "downsample_padding": 1,
+  "downsample_type": "resnet",
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 3,
@@ -26,11 +28,10 @@
   "out_channels": 3,
   "resnet_time_scale_shift": "scale_shift",
   "sample_size": 32,
-  "upsample_type": "resnet",
-  "downsample_type": "resnet",
   "time_embedding_type": "positional",
   "up_block_types": [
     "AttnUpBlock2D",
     "ResnetUpsampleBlock2D"
-  ]
+  ],
+  "upsample_type": "resnet"
 }

test_unet_class_cond/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74c24e2a295ea2c52edf3c967ee330cfafa2fef53f82c943dde61a77655dc125
+size 5145572