msluszniak commited on
Commit
7723873
·
verified ·
1 Parent(s): ec29487

Add the Vulkan fp16 exports

Browse files
.gitattributes CHANGED
@@ -34,4 +34,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  xnnpack/** filter=lfs diff=lfs merge=lfs -text
37
- mlx/** filter=lfs diff=lfs merge=lfs -text
 
 
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  xnnpack/** filter=lfs diff=lfs merge=lfs -text
37
+ mlx/** filter=lfs diff=lfs merge=lfs -textvulkan/duration_predictor_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
38
+ vulkan/text_encoder_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
39
+ vulkan/vector_estimator_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
40
+ vulkan/vocoder_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
vulkan/config.json ADDED
@@ -0,0 +1,203 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "supertonic3_vulkan_fp16",
4
+ "family": "supertonic",
5
+ "capabilities": [
6
+ "text-to-speech"
7
+ ],
8
+ "backend": "vulkan",
9
+ "license": "openrail",
10
+ "variants": [
11
+ {
12
+ "file": "duration_predictor_vulkan_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": true,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ null
23
+ ],
24
+ "dtype": "int64"
25
+ },
26
+ {
27
+ "shape": [
28
+ 1,
29
+ 8,
30
+ 16
31
+ ],
32
+ "dtype": "float32"
33
+ },
34
+ {
35
+ "shape": [
36
+ 1,
37
+ 1,
38
+ null
39
+ ],
40
+ "dtype": "float32"
41
+ }
42
+ ],
43
+ "outputs": [
44
+ {
45
+ "shape": [
46
+ 1
47
+ ],
48
+ "dtype": "float32"
49
+ }
50
+ ]
51
+ }
52
+ }
53
+ },
54
+ {
55
+ "file": "text_encoder_vulkan_fp16.pte",
56
+ "precision": "fp16",
57
+ "quantized": false,
58
+ "default": true,
59
+ "methods": {
60
+ "forward": {
61
+ "inputs": [
62
+ {
63
+ "shape": [
64
+ 1,
65
+ null
66
+ ],
67
+ "dtype": "int64"
68
+ },
69
+ {
70
+ "shape": [
71
+ 1,
72
+ 50,
73
+ 256
74
+ ],
75
+ "dtype": "float32"
76
+ },
77
+ {
78
+ "shape": [
79
+ 1,
80
+ 1,
81
+ null
82
+ ],
83
+ "dtype": "float32"
84
+ }
85
+ ],
86
+ "outputs": [
87
+ {
88
+ "shape": [
89
+ 1,
90
+ 256,
91
+ null
92
+ ],
93
+ "dtype": "float32"
94
+ }
95
+ ]
96
+ }
97
+ }
98
+ },
99
+ {
100
+ "file": "vector_estimator_vulkan_fp16.pte",
101
+ "precision": "fp16",
102
+ "quantized": false,
103
+ "default": true,
104
+ "methods": {
105
+ "forward": {
106
+ "inputs": [
107
+ {
108
+ "shape": [
109
+ 1,
110
+ 144,
111
+ null
112
+ ],
113
+ "dtype": "float32"
114
+ },
115
+ {
116
+ "shape": [
117
+ 1,
118
+ 256,
119
+ null
120
+ ],
121
+ "dtype": "float32"
122
+ },
123
+ {
124
+ "shape": [
125
+ 1,
126
+ 50,
127
+ 256
128
+ ],
129
+ "dtype": "float32"
130
+ },
131
+ {
132
+ "shape": [
133
+ 1,
134
+ 1,
135
+ null
136
+ ],
137
+ "dtype": "float32"
138
+ },
139
+ {
140
+ "shape": [
141
+ 1,
142
+ 1,
143
+ null
144
+ ],
145
+ "dtype": "float32"
146
+ },
147
+ {
148
+ "shape": [
149
+ 1
150
+ ],
151
+ "dtype": "float32"
152
+ },
153
+ {
154
+ "shape": [
155
+ 1
156
+ ],
157
+ "dtype": "float32"
158
+ }
159
+ ],
160
+ "outputs": [
161
+ {
162
+ "shape": [
163
+ 1,
164
+ 144,
165
+ null
166
+ ],
167
+ "dtype": "float32"
168
+ }
169
+ ]
170
+ }
171
+ }
172
+ },
173
+ {
174
+ "file": "vocoder_vulkan_fp16.pte",
175
+ "precision": "fp16",
176
+ "quantized": false,
177
+ "default": true,
178
+ "methods": {
179
+ "forward": {
180
+ "inputs": [
181
+ {
182
+ "shape": [
183
+ 1,
184
+ 144,
185
+ null
186
+ ],
187
+ "dtype": "float32"
188
+ }
189
+ ],
190
+ "outputs": [
191
+ {
192
+ "shape": [
193
+ 1,
194
+ null
195
+ ],
196
+ "dtype": "float32"
197
+ }
198
+ ]
199
+ }
200
+ }
201
+ }
202
+ ]
203
+ }
vulkan/duration_predictor_vulkan_fp16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72b835b121fe385b9c54d9c7ead471b5edb0fc31799044b196e9475493a9a1a6
3
+ size 1817346
vulkan/text_encoder_vulkan_fp16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0eb2ae90843a825a2508834e65767f2605aa7f9bef9cf67066eb4f34f51421cb
3
+ size 18148354
vulkan/vector_estimator_vulkan_fp16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37ffcf7966e15767ecddd2fa1f0365f5855f0a080a1ea209b48d15197f17b88c
3
+ size 128332162
vulkan/vocoder_vulkan_fp16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c84853b6e9635fb1d3b397cffdc4a19c063e5ce500e4b5494a4f5a1bbcfd43a
3
+ size 50746880