msluszniak commited on
Commit
280199d
·
verified ·
1 Parent(s): 907fff6

Add per-backend config

Browse files
Files changed (1) hide show
  1. mlx/config.json +26 -0
mlx/config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "ssdlite320_mobilenet_v3_large",
4
+ "family": "ssdlite",
5
+ "capabilities": ["object-detection"],
6
+ "backend": "mlx",
7
+ "license": "bsd-3-clause",
8
+ "variants": [
9
+ {
10
+ "file": "ssdlite320_mobilenet_v3_large_mlx_fp32.pte",
11
+ "precision": "fp32",
12
+ "quantized": false,
13
+ "default": true,
14
+ "methods": {
15
+ "forward": {
16
+ "inputs": [{ "shape": [3, 320, 320], "dtype": "float32" }],
17
+ "outputs": [
18
+ { "shape": [300, 4], "dtype": "float32" },
19
+ { "shape": [300], "dtype": "float32" },
20
+ { "shape": [300], "dtype": "float32" }
21
+ ]
22
+ }
23
+ }
24
+ }
25
+ ]
26
+ }