Yuanshi commited on
Commit
c87d952
·
verified ·
1 Parent(s): 1d5d021

Upload LinFusion

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. config.json +98 -83
README.md CHANGED
@@ -1,4 +1,5 @@
1
  ---
 
2
  tags:
3
  - pytorch_model_hub_mixin
4
  - model_hub_mixin
 
1
  ---
2
+ library_name: diffusers
3
  tags:
4
  - pytorch_model_hub_mixin
5
  - model_hub_mixin
config.json CHANGED
@@ -1,87 +1,102 @@
1
  {
2
- "_class_name": "StableMamba",
3
- "_diffusers_version": "0.25.0",
4
- "_name_or_path": "model_test",
5
  "modules_list": [
6
- [
7
- "down_blocks.0.attentions.0.transformer_blocks.0.attn1",
8
- 320,
9
- 8
10
- ],
11
- [
12
- "down_blocks.0.attentions.1.transformer_blocks.0.attn1",
13
- 320,
14
- 8
15
- ],
16
- [
17
- "down_blocks.1.attentions.0.transformer_blocks.0.attn1",
18
- 640,
19
- 8
20
- ],
21
- [
22
- "down_blocks.1.attentions.1.transformer_blocks.0.attn1",
23
- 640,
24
- 8
25
- ],
26
- [
27
- "down_blocks.2.attentions.0.transformer_blocks.0.attn1",
28
- 1280,
29
- 8
30
- ],
31
- [
32
- "down_blocks.2.attentions.1.transformer_blocks.0.attn1",
33
- 1280,
34
- 8
35
- ],
36
- [
37
- "up_blocks.1.attentions.0.transformer_blocks.0.attn1",
38
- 1280,
39
- 8
40
- ],
41
- [
42
- "up_blocks.1.attentions.1.transformer_blocks.0.attn1",
43
- 1280,
44
- 8
45
- ],
46
- [
47
- "up_blocks.1.attentions.2.transformer_blocks.0.attn1",
48
- 1280,
49
- 8
50
- ],
51
- [
52
- "up_blocks.2.attentions.0.transformer_blocks.0.attn1",
53
- 640,
54
- 8
55
- ],
56
- [
57
- "up_blocks.2.attentions.1.transformer_blocks.0.attn1",
58
- 640,
59
- 8
60
- ],
61
- [
62
- "up_blocks.2.attentions.2.transformer_blocks.0.attn1",
63
- 640,
64
- 8
65
- ],
66
- [
67
- "up_blocks.3.attentions.0.transformer_blocks.0.attn1",
68
- 320,
69
- 8
70
- ],
71
- [
72
- "up_blocks.3.attentions.1.transformer_blocks.0.attn1",
73
- 320,
74
- 8
75
- ],
76
- [
77
- "up_blocks.3.attentions.2.transformer_blocks.0.attn1",
78
- 320,
79
- 8
80
- ],
81
- [
82
- "mid_block.attentions.0.transformer_blocks.0.attn1",
83
- 1280,
84
- 8
85
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
86
  ]
87
  }
 
1
  {
2
+ "_class_name": "LinFusion",
3
+ "_diffusers_version": "0.30.0",
 
4
  "modules_list": [
5
+ {
6
+ "dim_n": 320,
7
+ "heads": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
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": 8,
98
+ "module_name": "mid_block.attentions.0.transformer_blocks.0.attn1",
99
+ "projection_mid_dim": null
100
+ }
101
  ]
102
  }