patrickvonplaten dg845 commited on
Commit
5f462e4
1 Parent(s): 45d6e0c

Upload new cd_imagenet64_l2 consistency model checkpoint (fix mid block attention group norm bug) (#1)

Browse files

- Upload new cd_imagenet64_l2 consistency model checkpoint (fix mid block attention group norm bug) (b3163073330d94da85ef6a1858a668afeda394c5)


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

model_index.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "_class_name": "ConsistencyModelPipeline",
3
- "_diffusers_version": "0.17.0.dev0",
 
4
  "scheduler": [
5
  "diffusers",
6
  "CMStochasticIterativeScheduler"
 
1
  {
2
  "_class_name": "ConsistencyModelPipeline",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "_name_or_path": "cd_imagenet64_l2",
5
  "scheduler": [
6
  "diffusers",
7
  "CMStochasticIterativeScheduler"
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "CMStochasticIterativeScheduler",
3
- "_diffusers_version": "0.17.0.dev0",
4
  "clip_denoised": true,
5
  "num_train_timesteps": 40,
6
  "rho": 7.0,
 
1
  {
2
  "_class_name": "CMStochasticIterativeScheduler",
3
+ "_diffusers_version": "0.21.0.dev0",
4
  "clip_denoised": true,
5
  "num_train_timesteps": 40,
6
  "rho": 7.0,
unet/config.json CHANGED
@@ -1,10 +1,11 @@
1
  {
2
  "_class_name": "UNet2DModel",
3
- "_diffusers_version": "0.17.0.dev0",
4
- "_name_or_path": "ayushtues/consistency_models",
5
  "act_fn": "silu",
6
  "add_attention": true,
7
  "attention_head_dim": 64,
 
8
  "block_out_channels": [
9
  192,
10
  384,
@@ -20,6 +21,7 @@
20
  "AttnDownBlock2D"
21
  ],
22
  "downsample_padding": 1,
 
23
  "flip_sin_to_cos": true,
24
  "freq_shift": 0,
25
  "in_channels": 3,
@@ -32,12 +34,11 @@
32
  "resnet_time_scale_shift": "scale_shift",
33
  "sample_size": 64,
34
  "time_embedding_type": "positional",
35
- "upsample_type": "resnet",
36
- "downsample_type": "resnet",
37
  "up_block_types": [
38
  "AttnUpBlock2D",
39
  "AttnUpBlock2D",
40
  "AttnUpBlock2D",
41
  "ResnetUpsampleBlock2D"
42
- ]
 
43
  }
 
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "_name_or_path": "cd_imagenet64_l2/unet",
5
  "act_fn": "silu",
6
  "add_attention": true,
7
  "attention_head_dim": 64,
8
+ "attn_norm_num_groups": 32,
9
  "block_out_channels": [
10
  192,
11
  384,
 
21
  "AttnDownBlock2D"
22
  ],
23
  "downsample_padding": 1,
24
+ "downsample_type": "resnet",
25
  "flip_sin_to_cos": true,
26
  "freq_shift": 0,
27
  "in_channels": 3,
 
34
  "resnet_time_scale_shift": "scale_shift",
35
  "sample_size": 64,
36
  "time_embedding_type": "positional",
 
 
37
  "up_block_types": [
38
  "AttnUpBlock2D",
39
  "AttnUpBlock2D",
40
  "AttnUpBlock2D",
41
  "ResnetUpsampleBlock2D"
42
+ ],
43
+ "upsample_type": "resnet"
44
  }
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9702bf70b1e7cbe67b1da6d829e90c469f78c0592b1760170dbfb9d370f5695c
3
+ size 1183665484