Hugging Face's logo

software-mansion
/
react-native-executorch-bk-sdm-tiny

ExecuTorch
Model card Files
xet
 Community
msluszniak commited on
Commit
cda6002
·
verified ·
1 Parent(s): 1518c5a

Add spec-compliant config.json files

Browse files
Files changed (1) hide show
  1. xnnpack/config.json +172 -5
xnnpack/config.json CHANGED
@@ -12,31 +12,198 @@
12
  "file": "bk_sdm_tiny_text_encoder_xnnpack_fp32.pte",
13
  "precision": "fp32",
14
  "quantized": false,
15
- "default": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  },
17
  {
18
  "file": "bk_sdm_tiny_unet_256_xnnpack_fp32.pte",
19
  "precision": "fp32",
20
  "quantized": false,
21
- "default": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  },
23
  {
24
  "file": "bk_sdm_tiny_unet_xnnpack_fp32.pte",
25
  "precision": "fp32",
26
  "quantized": false,
27
- "default": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  },
29
  {
30
  "file": "bk_sdm_tiny_vae_256_xnnpack_fp32.pte",
31
  "precision": "fp32",
32
  "quantized": false,
33
- "default": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  },
35
  {
36
  "file": "bk_sdm_tiny_vae_xnnpack_fp32.pte",
37
  "precision": "fp32",
38
  "quantized": false,
39
- "default": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  }
41
  ]
42
  }
 
12
  "file": "bk_sdm_tiny_text_encoder_xnnpack_fp32.pte",
13
  "precision": "fp32",
14
  "quantized": false,
15
+ "default": true,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ 77
23
+ ],
24
+ "dtype": "int64"
25
+ },
26
+ {
27
+ "shape": [
28
+ 1,
29
+ 77
30
+ ],
31
+ "dtype": "int64"
32
+ }
33
+ ],
34
+ "outputs": [
35
+ {
36
+ "shape": [
37
+ 1,
38
+ 77,
39
+ 768
40
+ ],
41
+ "dtype": "float32"
42
+ },
43
+ {
44
+ "shape": [
45
+ 1,
46
+ 768
47
+ ],
48
+ "dtype": "float32"
49
+ }
50
+ ]
51
+ }
52
+ }
53
  },
54
  {
55
  "file": "bk_sdm_tiny_unet_256_xnnpack_fp32.pte",
56
  "precision": "fp32",
57
  "quantized": false,
58
+ "default": false,
59
+ "methods": {
60
+ "forward": {
61
+ "inputs": [
62
+ {
63
+ "shape": [
64
+ 2,
65
+ 4,
66
+ 32,
67
+ 32
68
+ ],
69
+ "dtype": "float32"
70
+ },
71
+ {
72
+ "shape": [
73
+ 1
74
+ ],
75
+ "dtype": "int64"
76
+ },
77
+ {
78
+ "shape": [
79
+ 2,
80
+ 77,
81
+ 768
82
+ ],
83
+ "dtype": "float32"
84
+ }
85
+ ],
86
+ "outputs": [
87
+ {
88
+ "shape": [
89
+ 2,
90
+ 4,
91
+ 32,
92
+ 32
93
+ ],
94
+ "dtype": "float32"
95
+ }
96
+ ]
97
+ }
98
+ }
99
  },
100
  {
101
  "file": "bk_sdm_tiny_unet_xnnpack_fp32.pte",
102
  "precision": "fp32",
103
  "quantized": false,
104
+ "default": false,
105
+ "methods": {
106
+ "forward": {
107
+ "inputs": [
108
+ {
109
+ "shape": [
110
+ 2,
111
+ 4,
112
+ 64,
113
+ 64
114
+ ],
115
+ "dtype": "float32"
116
+ },
117
+ {
118
+ "shape": [
119
+ 1
120
+ ],
121
+ "dtype": "int64"
122
+ },
123
+ {
124
+ "shape": [
125
+ 2,
126
+ 77,
127
+ 768
128
+ ],
129
+ "dtype": "float32"
130
+ }
131
+ ],
132
+ "outputs": [
133
+ {
134
+ "shape": [
135
+ 2,
136
+ 4,
137
+ 64,
138
+ 64
139
+ ],
140
+ "dtype": "float32"
141
+ }
142
+ ]
143
+ }
144
+ }
145
  },
146
  {
147
  "file": "bk_sdm_tiny_vae_256_xnnpack_fp32.pte",
148
  "precision": "fp32",
149
  "quantized": false,
150
+ "default": false,
151
+ "methods": {
152
+ "forward": {
153
+ "inputs": [
154
+ {
155
+ "shape": [
156
+ 1,
157
+ 4,
158
+ 32,
159
+ 32
160
+ ],
161
+ "dtype": "float32"
162
+ }
163
+ ],
164
+ "outputs": [
165
+ {
166
+ "shape": [
167
+ 256,
168
+ 256,
169
+ 3
170
+ ],
171
+ "dtype": "float32"
172
+ }
173
+ ]
174
+ }
175
+ }
176
  },
177
  {
178
  "file": "bk_sdm_tiny_vae_xnnpack_fp32.pte",
179
  "precision": "fp32",
180
  "quantized": false,
181
+ "default": false,
182
+ "methods": {
183
+ "forward": {
184
+ "inputs": [
185
+ {
186
+ "shape": [
187
+ 1,
188
+ 4,
189
+ 64,
190
+ 64
191
+ ],
192
+ "dtype": "float32"
193
+ }
194
+ ],
195
+ "outputs": [
196
+ {
197
+ "shape": [
198
+ 512,
199
+ 512,
200
+ 3
201
+ ],
202
+ "dtype": "float32"
203
+ }
204
+ ]
205
+ }
206
+ }
207
  }
208
  ]
209
  }