Hugging Face's logo

software-mansion
/
react-native-executorch-all-MiniLM-L6-v2

ExecuTorch
Model card Files
xet
 Community
2
msluszniak commited on
Commit
2ef7cc5
·
verified ·
1 Parent(s): afd9a50

Add per-backend config

Browse files
Files changed (1) hide show
  1. coreml/config.json +82 -0
coreml/config.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "all_minilm_l6_v2",
4
+ "family": "sbert",
5
+ "capabilities": [
6
+ "text-embedding"
7
+ ],
8
+ "backend": "coreml",
9
+ "license": "apache-2.0",
10
+ "variants": [
11
+ {
12
+ "file": "all_minilm_l6_v2_coreml_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": true,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ 254
23
+ ],
24
+ "dtype": "int64"
25
+ },
26
+ {
27
+ "shape": [
28
+ 1,
29
+ 254
30
+ ],
31
+ "dtype": "int64"
32
+ }
33
+ ],
34
+ "outputs": [
35
+ {
36
+ "shape": [
37
+ 1,
38
+ 384
39
+ ],
40
+ "dtype": "float32"
41
+ }
42
+ ]
43
+ }
44
+ }
45
+ },
46
+ {
47
+ "file": "all_minilm_l6_v2_coreml_fp32.pte",
48
+ "precision": "fp32",
49
+ "quantized": false,
50
+ "default": false,
51
+ "methods": {
52
+ "forward": {
53
+ "inputs": [
54
+ {
55
+ "shape": [
56
+ 1,
57
+ 254
58
+ ],
59
+ "dtype": "int64"
60
+ },
61
+ {
62
+ "shape": [
63
+ 1,
64
+ 254
65
+ ],
66
+ "dtype": "int64"
67
+ }
68
+ ],
69
+ "outputs": [
70
+ {
71
+ "shape": [
72
+ 1,
73
+ 384
74
+ ],
75
+ "dtype": "float32"
76
+ }
77
+ ]
78
+ }
79
+ }
80
+ }
81
+ ]
82
+ }