patrickvonplaten commited on
Commit
c073669
1 Parent(s): 207ff98

convert weights

Browse files
unet/config.json CHANGED
@@ -3,12 +3,6 @@
3
  "_diffusers_version": "0.0.4",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
- "block_channels": [
7
- 320,
8
- 640,
9
- 1280,
10
- 1280
11
- ],
12
  "block_out_channels": [
13
  320,
14
  640,
@@ -16,30 +10,23 @@
16
  1280
17
  ],
18
  "center_input_sample": false,
19
- "down_blocks": [
20
  "CrossAttnDownBlock2D",
21
  "CrossAttnDownBlock2D",
22
  "CrossAttnDownBlock2D",
23
  "DownBlock2D"
24
  ],
25
  "downsample_padding": 1,
26
- "downscale_freq_shift": 0,
27
  "flip_sin_to_cos": true,
28
  "freq_shift": 0,
29
- "image_size": null,
30
  "in_channels": 4,
31
  "layers_per_block": 2,
32
  "mid_block_scale_factor": 1,
33
  "norm_eps": 1e-05,
34
  "norm_num_groups": 32,
35
- "num_head_channels": 8,
36
- "num_res_blocks": 2,
37
  "out_channels": 4,
38
- "resnet_act_fn": "silu",
39
- "resnet_eps": 1e-05,
40
- "resnet_num_groups": 30,
41
  "sample_size": 32,
42
- "up_blocks": [
43
  "UpBlock2D",
44
  "CrossAttnUpBlock2D",
45
  "CrossAttnUpBlock2D",
 
3
  "_diffusers_version": "0.0.4",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
 
 
 
 
 
 
6
  "block_out_channels": [
7
  320,
8
  640,
 
10
  1280
11
  ],
12
  "center_input_sample": false,
13
+ "down_block_types": [
14
  "CrossAttnDownBlock2D",
15
  "CrossAttnDownBlock2D",
16
  "CrossAttnDownBlock2D",
17
  "DownBlock2D"
18
  ],
19
  "downsample_padding": 1,
 
20
  "flip_sin_to_cos": true,
21
  "freq_shift": 0,
 
22
  "in_channels": 4,
23
  "layers_per_block": 2,
24
  "mid_block_scale_factor": 1,
25
  "norm_eps": 1e-05,
26
  "norm_num_groups": 32,
 
 
27
  "out_channels": 4,
 
 
 
28
  "sample_size": 32,
29
+ "up_block_types": [
30
  "UpBlock2D",
31
  "CrossAttnUpBlock2D",
32
  "CrossAttnUpBlock2D",
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7fc9295026da8bcbbf4eedfe41714d4691e954ebcfbed224010c8bcd5aaea5e2
3
- size 3489480869
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61459453cb8e411842ff1c109da94129f94349d46e579dbec303449aa74f07bf
3
+ size 3489472805