Yuanshi commited on
Commit
4ed4189
1 Parent(s): 9b89aa4

Upload LinFusion

Browse files
Files changed (2) hide show
  1. config.json +97 -81
  2. diffusion_pytorch_model.safetensors +2 -2
config.json CHANGED
@@ -1,86 +1,102 @@
1
  {
2
- "_class_name": "StableMamba",
3
  "_diffusers_version": "0.30.0",
4
  "modules_list": [
5
- [
6
- "down_blocks.0.attentions.0.transformer_blocks.0.attn1",
7
- 320,
8
- 5
9
- ],
10
- [
11
- "down_blocks.0.attentions.1.transformer_blocks.0.attn1",
12
- 320,
13
- 5
14
- ],
15
- [
16
- "down_blocks.1.attentions.0.transformer_blocks.0.attn1",
17
- 640,
18
- 10
19
- ],
20
- [
21
- "down_blocks.1.attentions.1.transformer_blocks.0.attn1",
22
- 640,
23
- 10
24
- ],
25
- [
26
- "down_blocks.2.attentions.0.transformer_blocks.0.attn1",
27
- 1280,
28
- 20
29
- ],
30
- [
31
- "down_blocks.2.attentions.1.transformer_blocks.0.attn1",
32
- 1280,
33
- 20
34
- ],
35
- [
36
- "up_blocks.1.attentions.0.transformer_blocks.0.attn1",
37
- 1280,
38
- 20
39
- ],
40
- [
41
- "up_blocks.1.attentions.1.transformer_blocks.0.attn1",
42
- 1280,
43
- 20
44
- ],
45
- [
46
- "up_blocks.1.attentions.2.transformer_blocks.0.attn1",
47
- 1280,
48
- 20
49
- ],
50
- [
51
- "up_blocks.2.attentions.0.transformer_blocks.0.attn1",
52
- 640,
53
- 10
54
- ],
55
- [
56
- "up_blocks.2.attentions.1.transformer_blocks.0.attn1",
57
- 640,
58
- 10
59
- ],
60
- [
61
- "up_blocks.2.attentions.2.transformer_blocks.0.attn1",
62
- 640,
63
- 10
64
- ],
65
- [
66
- "up_blocks.3.attentions.0.transformer_blocks.0.attn1",
67
- 320,
68
- 5
69
- ],
70
- [
71
- "up_blocks.3.attentions.1.transformer_blocks.0.attn1",
72
- 320,
73
- 5
74
- ],
75
- [
76
- "up_blocks.3.attentions.2.transformer_blocks.0.attn1",
77
- 320,
78
- 5
79
- ],
80
- [
81
- "mid_block.attentions.0.transformer_blocks.0.attn1",
82
- 1280,
83
- 20
84
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
85
  ]
86
  }
 
1
  {
2
+ "_class_name": "LinFusion",
3
  "_diffusers_version": "0.30.0",
4
  "modules_list": [
5
+ {
6
+ "dim_n": 320,
7
+ "heads": 5,
8
+ "module_name": "down_blocks.0.attentions.0.transformer_blocks.0.attn1",
9
+ "projection_mid_dim": null
10
+ },
11
+ {
12
+ "dim_n": 320,
13
+ "heads": 5,
14
+ "module_name": "down_blocks.0.attentions.1.transformer_blocks.0.attn1",
15
+ "projection_mid_dim": null
16
+ },
17
+ {
18
+ "dim_n": 640,
19
+ "heads": 10,
20
+ "module_name": "down_blocks.1.attentions.0.transformer_blocks.0.attn1",
21
+ "projection_mid_dim": null
22
+ },
23
+ {
24
+ "dim_n": 640,
25
+ "heads": 10,
26
+ "module_name": "down_blocks.1.attentions.1.transformer_blocks.0.attn1",
27
+ "projection_mid_dim": null
28
+ },
29
+ {
30
+ "dim_n": 1280,
31
+ "heads": 20,
32
+ "module_name": "down_blocks.2.attentions.0.transformer_blocks.0.attn1",
33
+ "projection_mid_dim": null
34
+ },
35
+ {
36
+ "dim_n": 1280,
37
+ "heads": 20,
38
+ "module_name": "down_blocks.2.attentions.1.transformer_blocks.0.attn1",
39
+ "projection_mid_dim": null
40
+ },
41
+ {
42
+ "dim_n": 1280,
43
+ "heads": 20,
44
+ "module_name": "up_blocks.1.attentions.0.transformer_blocks.0.attn1",
45
+ "projection_mid_dim": null
46
+ },
47
+ {
48
+ "dim_n": 1280,
49
+ "heads": 20,
50
+ "module_name": "up_blocks.1.attentions.1.transformer_blocks.0.attn1",
51
+ "projection_mid_dim": null
52
+ },
53
+ {
54
+ "dim_n": 1280,
55
+ "heads": 20,
56
+ "module_name": "up_blocks.1.attentions.2.transformer_blocks.0.attn1",
57
+ "projection_mid_dim": null
58
+ },
59
+ {
60
+ "dim_n": 640,
61
+ "heads": 10,
62
+ "module_name": "up_blocks.2.attentions.0.transformer_blocks.0.attn1",
63
+ "projection_mid_dim": null
64
+ },
65
+ {
66
+ "dim_n": 640,
67
+ "heads": 10,
68
+ "module_name": "up_blocks.2.attentions.1.transformer_blocks.0.attn1",
69
+ "projection_mid_dim": null
70
+ },
71
+ {
72
+ "dim_n": 640,
73
+ "heads": 10,
74
+ "module_name": "up_blocks.2.attentions.2.transformer_blocks.0.attn1",
75
+ "projection_mid_dim": null
76
+ },
77
+ {
78
+ "dim_n": 320,
79
+ "heads": 5,
80
+ "module_name": "up_blocks.3.attentions.0.transformer_blocks.0.attn1",
81
+ "projection_mid_dim": null
82
+ },
83
+ {
84
+ "dim_n": 320,
85
+ "heads": 5,
86
+ "module_name": "up_blocks.3.attentions.1.transformer_blocks.0.attn1",
87
+ "projection_mid_dim": null
88
+ },
89
+ {
90
+ "dim_n": 320,
91
+ "heads": 5,
92
+ "module_name": "up_blocks.3.attentions.2.transformer_blocks.0.attn1",
93
+ "projection_mid_dim": null
94
+ },
95
+ {
96
+ "dim_n": 1280,
97
+ "heads": 20,
98
+ "module_name": "mid_block.attentions.0.transformer_blocks.0.attn1",
99
+ "projection_mid_dim": null
100
+ }
101
  ]
102
  }
diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ccc544c938633d085672f43fbb5fa46cd5d1bb068c07adbbb74e2872144e49e
3
- size 198494384
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46602c8291412a32760280b77650d2324fee8e454c69c3931911d6a5fb5e317e
3
+ size 198494656