msluszniak commited on
Commit
ca4e8aa
·
1 Parent(s): eba2f5b

Add Vulkan fp16 exports

Browse files
vulkan/config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "style_transfer_candy",
4
+ "family": "style_transfer",
5
+ "capabilities": [
6
+ "style-transfer"
7
+ ],
8
+ "backend": "vulkan",
9
+ "license": "bsd-3-clause",
10
+ "variants": [
11
+ {
12
+ "file": "style_transfer_candy_vulkan_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": true,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ 3,
23
+ 640,
24
+ 640
25
+ ],
26
+ "dtype": "float32"
27
+ }
28
+ ],
29
+ "outputs": [
30
+ {
31
+ "shape": [
32
+ 1,
33
+ 3,
34
+ 640,
35
+ 640
36
+ ],
37
+ "dtype": "float32"
38
+ }
39
+ ]
40
+ }
41
+ }
42
+ },
43
+ {
44
+ "file": "style_transfer_candy_vulkan_fp32.pte",
45
+ "precision": "fp32",
46
+ "quantized": false,
47
+ "default": false,
48
+ "methods": {
49
+ "forward": {
50
+ "inputs": [
51
+ {
52
+ "shape": [
53
+ 1,
54
+ 3,
55
+ 640,
56
+ 640
57
+ ],
58
+ "dtype": "float32"
59
+ }
60
+ ],
61
+ "outputs": [
62
+ {
63
+ "shape": [
64
+ 1,
65
+ 3,
66
+ 640,
67
+ 640
68
+ ],
69
+ "dtype": "float32"
70
+ }
71
+ ]
72
+ }
73
+ }
74
+ }
75
+ ]
76
+ }
vulkan/style_transfer_candy_vulkan_fp16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5eb4885ec0a816872986a3cb1dbce9dbca0a3a952334e511ae2b708defe84007
3
+ size 3418114