msluszniak commited on
Commit
2371aa4
·
verified ·
1 Parent(s): 37f736a

Add backend config (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. x/mlx/config.json +62 -0
x/mlx/config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "fast_sam_x",
4
+ "family": "fastsam",
5
+ "capabilities": [
6
+ "instance-segmentation"
7
+ ],
8
+ "backend": "mlx",
9
+ "license": "agpl-3.0",
10
+ "variants": [
11
+ {
12
+ "file": "fast_sam_x_mlx_fp32.pte",
13
+ "precision": "fp32",
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
+ 300,
33
+ 4
34
+ ],
35
+ "dtype": "float32"
36
+ },
37
+ {
38
+ "shape": [
39
+ 300
40
+ ],
41
+ "dtype": "float32"
42
+ },
43
+ {
44
+ "shape": [
45
+ 300
46
+ ],
47
+ "dtype": "float32"
48
+ },
49
+ {
50
+ "shape": [
51
+ 300,
52
+ 160,
53
+ 160
54
+ ],
55
+ "dtype": "float32"
56
+ }
57
+ ]
58
+ }
59
+ }
60
+ }
61
+ ]
62
+ }