msluszniak commited on
Commit
06d8355
·
verified ·
1 Parent(s): ae2e776

Add XNNPACK int8 (8da8w) variant

Browse files
.gitattributes CHANGED
@@ -39,3 +39,4 @@ xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte filter=lfs diff=lfs merge=lf
39
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
40
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
41
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte filter=lfs diff=lfs merge=lfs -text
 
 
39
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
40
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
41
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte filter=lfs diff=lfs merge=lfs -text
42
+ xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -20,16 +20,19 @@ Upstream model: [multi-qa-mpnet-base-dot-v1](https://huggingface.co/sentence-tra
20
  | `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte` | vulkan | int8 |
21
  | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` | xnnpack | 8da4w |
22
  | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte` | xnnpack | fp32 |
 
23
 
24
  ## XNNPACK int8
25
 
26
- `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` is 8da4w (int8 dynamic activations, int4 weights,
27
- group 32) on the Linear layers, with an int8 word-embedding table. SQuAD v1.1 dev
28
- passage retrieval (2000 questions against their 315 paragraphs), dot product:
 
29
 
30
  | | top-1 | MRR |
31
  | --- | --- | --- |
32
  | fp32 | 64.60 | 0.7645 |
 
33
  | 8da4w | 63.80 | 0.7578 |
34
 
35
  ## Repository structure
@@ -41,9 +44,10 @@ tokenizer_config.json 363 B
41
  vulkan/config.json 1.6 kB
42
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte 208 MB
43
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte 127 MB
44
- xnnpack/config.json 1.6 kB
45
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte 69.1 MB
46
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte 416 MB
 
47
  ```
48
 
49
  ## Compatibility
 
20
  | `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte` | vulkan | int8 |
21
  | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` | xnnpack | 8da4w |
22
  | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte` | xnnpack | fp32 |
23
+ | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte` | xnnpack | int8 |
24
 
25
  ## XNNPACK int8
26
 
27
+ `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte` is 8da8w (int8 dynamic activations, int8 per-channel
28
+ weights) and `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` is 8da4w (int4 weights, group 32), both on the
29
+ Linear layers with an int8 word-embedding table. SQuAD v1.1 dev passage retrieval (2000
30
+ questions against their 315 paragraphs), dot product:
31
 
32
  | | top-1 | MRR |
33
  | --- | --- | --- |
34
  | fp32 | 64.60 | 0.7645 |
35
+ | int8 | 65.00 | 0.7671 |
36
  | 8da4w | 63.80 | 0.7578 |
37
 
38
  ## Repository structure
 
44
  vulkan/config.json 1.6 kB
45
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte 208 MB
46
  vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte 127 MB
47
+ xnnpack/config.json 2.2 kB
48
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte 69.1 MB
49
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte 416 MB
50
+ xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte 105 MB
51
  ```
52
 
53
  ## Compatibility
xnnpack/config.json CHANGED
@@ -41,6 +41,39 @@
41
  }
42
  }
43
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  {
45
  "file": "multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte",
46
  "precision": "8da4w",
 
41
  }
42
  }
43
  },
44
+ {
45
+ "file": "multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte",
46
+ "precision": "int8",
47
+ "methods": {
48
+ "forward": {
49
+ "inputs": [
50
+ {
51
+ "shape": [
52
+ 1,
53
+ 510
54
+ ],
55
+ "dtype": "int64"
56
+ },
57
+ {
58
+ "shape": [
59
+ 1,
60
+ 510
61
+ ],
62
+ "dtype": "int64"
63
+ }
64
+ ],
65
+ "outputs": [
66
+ {
67
+ "shape": [
68
+ 1,
69
+ 768
70
+ ],
71
+ "dtype": "float32"
72
+ }
73
+ ]
74
+ }
75
+ }
76
+ },
77
  {
78
  "file": "multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte",
79
  "precision": "8da4w",
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_int8.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77fc6dfe6490d401dc0e5d3556251560f3c2c2d3095ba8ac795ead81c69d3a5c
3
+ size 109965568