UmerHA commited on
Commit
3cdec90
·
verified ·
1 Parent(s): 7104929

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +14 -19
  2. diffusion_pytorch_model.safetensors +2 -2
config.json CHANGED
@@ -4,14 +4,12 @@
4
  "attention_head_dim": [
5
  1,
6
  1,
7
- 2,
8
  2
9
  ],
10
  "block_out_channels": [
11
- 4,
12
- 8,
13
- 16,
14
- 16
15
  ],
16
  "channels_base": {
17
  "down - out": [
@@ -23,16 +21,10 @@
23
  640,
24
  640,
25
  1280,
26
- 1280,
27
- 1280,
28
- 1280,
29
  1280
30
  ],
31
  "mid - out": 1280,
32
  "up - in": [
33
- 1280,
34
- 1280,
35
- 1280,
36
  1280,
37
  1280,
38
  1280,
@@ -52,18 +44,21 @@
52
  96,
53
  256
54
  ],
55
- "cross_attention_dim": 1024,
56
  "down_block_types": [
 
57
  "CrossAttnDownBlock2D",
58
- "CrossAttnDownBlock2D",
59
- "CrossAttnDownBlock2D",
60
- "DownBlock2D"
61
  ],
62
  "learn_time_embedding": true,
63
- "norm_num_groups": 4,
64
- "sample_size": 64,
65
  "time_embedding_dim": 1280,
66
  "time_embedding_input_dim": 320,
67
- "transformer_layers_per_block": 1,
68
- "upcast_attention": false
 
 
 
 
69
  }
 
4
  "attention_head_dim": [
5
  1,
6
  1,
 
7
  2
8
  ],
9
  "block_out_channels": [
10
+ 32,
11
+ 64,
12
+ 128
 
13
  ],
14
  "channels_base": {
15
  "down - out": [
 
21
  640,
22
  640,
23
  1280,
 
 
 
24
  1280
25
  ],
26
  "mid - out": 1280,
27
  "up - in": [
 
 
 
28
  1280,
29
  1280,
30
  1280,
 
44
  96,
45
  256
46
  ],
47
+ "cross_attention_dim": 2048,
48
  "down_block_types": [
49
+ "DownBlock2D",
50
  "CrossAttnDownBlock2D",
51
+ "CrossAttnDownBlock2D"
 
 
52
  ],
53
  "learn_time_embedding": true,
54
+ "norm_num_groups": 32,
55
+ "sample_size": 128,
56
  "time_embedding_dim": 1280,
57
  "time_embedding_input_dim": 320,
58
+ "transformer_layers_per_block": [
59
+ 1,
60
+ 2,
61
+ 10
62
+ ],
63
+ "upcast_attention": null
64
  }
diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c82d4adb3e0352feae9ef4f2292cc42eba97b993c91a3b5cf5c19a4308606547
3
- size 56790464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:085253593fd3dbaa0f2b656a676156f01897082b6d543aa1ff7fd80165a190fd
3
+ size 169809408