patrickvonplaten commited on
Commit
5d6cca4
1 Parent(s): a57626b
generated_image_pipeline.png CHANGED
generated_image_unrolled.png CHANGED
unet/config.json CHANGED
@@ -6,30 +6,36 @@
6
  4,
7
  2
8
  ],
9
- "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D"],
10
- "up_blocks": ["UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
11
- "down_block_input_channels": [224, 224, 448, 672],
12
- "down_block_output_channels": [224, 448, 672, 896],
13
- "context_dim": null,
 
 
14
  "conv_resample": true,
15
- "dims": 2,
 
 
 
 
 
 
 
 
16
  "dropout": 0,
 
17
  "image_size": 64,
18
  "in_channels": 3,
19
- "legacy": true,
20
- "n_embed": null,
21
- "num_classes": null,
22
  "num_head_channels": 32,
23
- "num_heads": -1,
24
- "num_heads_upsample": -1,
25
  "num_res_blocks": 2,
26
  "out_channels": 3,
27
- "resblock_updown": false,
28
- "transformer_depth": 1,
29
- "use_checkpoint": false,
30
- "use_fp16": false,
31
- "use_new_attention_order": false,
32
- "use_scale_shift_norm": false,
33
- "use_spatial_transformer": false,
34
- "ldm": true
35
  }
 
6
  4,
7
  2
8
  ],
9
+ "attn_resolutions": null,
10
+ "block_channels": [
11
+ 224,
12
+ 448,
13
+ 672,
14
+ 896
15
+ ],
16
  "conv_resample": true,
17
+ "ddpm": false,
18
+ "down_blocks": [
19
+ "UNetResDownBlock2D",
20
+ "UNetResAttnDownBlock2D",
21
+ "UNetResAttnDownBlock2D",
22
+ "UNetResAttnDownBlock2D"
23
+ ],
24
+ "downsample_padding": 1,
25
+ "downscale_freq_shift": 0,
26
  "dropout": 0,
27
+ "flip_sin_to_cos": true,
28
  "image_size": 64,
29
  "in_channels": 3,
30
+ "ldm": true,
31
+ "name_or_path": "fusing/latent-diffusion-celeba-256",
 
32
  "num_head_channels": 32,
 
 
33
  "num_res_blocks": 2,
34
  "out_channels": 3,
35
+ "up_blocks": [
36
+ "UNetResAttnUpBlock2D",
37
+ "UNetResAttnUpBlock2D",
38
+ "UNetResAttnUpBlock2D",
39
+ "UNetResUpBlock2D"
40
+ ]
 
 
41
  }
unet/diffusion_model.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7b655ee0d741c2de23be13d7031c8365b7c17f61b5921a42d1173e1e20d48067
3
- size 1096382177
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57ec600227273ef7436e8447f654ecdd24f0a7543350456c0ee96aa72212fddd
3
+ size 2192783233