msluszniak commited on
Commit
75c4d05
·
verified ·
1 Parent(s): 89e6949

Add XNNPACK int8 (8da8w) variant

Browse files
.gitattributes CHANGED
@@ -39,3 +39,4 @@ xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
39
  vulkan/all_mpnet_base_v2_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
40
  vulkan/all_mpnet_base_v2_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
41
  xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte filter=lfs diff=lfs merge=lfs -text
 
 
39
  vulkan/all_mpnet_base_v2_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
40
  vulkan/all_mpnet_base_v2_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
41
  xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte filter=lfs diff=lfs merge=lfs -text
42
+ xnnpack/all_mpnet_base_v2_xnnpack_int8.pte filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -20,16 +20,19 @@ Upstream model: [all-mpnet-base-v2](https://huggingface.co/sentence-transformers
20
  | `vulkan/all_mpnet_base_v2_vulkan_int8.pte` | vulkan | int8 |
21
  | `xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte` | xnnpack | 8da4w |
22
  | `xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte` | xnnpack | fp32 |
 
23
 
24
  ## XNNPACK int8
25
 
26
- `xnnpack/all_mpnet_base_v2_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 | 65.05 | 0.7631 |
 
33
  | 8da4w | 64.60 | 0.7606 |
34
 
35
  ## Repository structure
@@ -43,7 +46,8 @@ vulkan/all_mpnet_base_v2_vulkan_int8.pte 127 MB
43
  vulkan/config.json 1.5 kB
44
  xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte 69.1 MB
45
  xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte 416 MB
46
- xnnpack/config.json 1.5 kB
 
47
  ```
48
 
49
  ## Compatibility
 
20
  | `vulkan/all_mpnet_base_v2_vulkan_int8.pte` | vulkan | int8 |
21
  | `xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte` | xnnpack | 8da4w |
22
  | `xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte` | xnnpack | fp32 |
23
+ | `xnnpack/all_mpnet_base_v2_xnnpack_int8.pte` | xnnpack | int8 |
24
 
25
  ## XNNPACK int8
26
 
27
+ `xnnpack/all_mpnet_base_v2_xnnpack_int8.pte` is 8da8w (int8 dynamic activations, int8 per-channel
28
+ weights) and `xnnpack/all_mpnet_base_v2_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 | 65.05 | 0.7631 |
35
+ | int8 | 65.00 | 0.7633 |
36
  | 8da4w | 64.60 | 0.7606 |
37
 
38
  ## Repository structure
 
46
  vulkan/config.json 1.5 kB
47
  xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte 69.1 MB
48
  xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte 416 MB
49
+ xnnpack/all_mpnet_base_v2_xnnpack_int8.pte 105 MB
50
+ xnnpack/config.json 2.2 kB
51
  ```
52
 
53
  ## Compatibility
xnnpack/all_mpnet_base_v2_xnnpack_int8.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61befa934901b544c665ec90d178ce4b1bb2ac1a772bb627df97face7323c2b9
3
+ size 109965568
xnnpack/config.json CHANGED
@@ -41,6 +41,39 @@
41
  }
42
  }
43
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  {
45
  "file": "all_mpnet_base_v2_xnnpack_8da4w.pte",
46
  "precision": "8da4w",
 
41
  }
42
  }
43
  },
44
+ {
45
+ "file": "all_mpnet_base_v2_xnnpack_int8.pte",
46
+ "precision": "int8",
47
+ "methods": {
48
+ "forward": {
49
+ "inputs": [
50
+ {
51
+ "shape": [
52
+ 1,
53
+ 382
54
+ ],
55
+ "dtype": "int64"
56
+ },
57
+ {
58
+ "shape": [
59
+ 1,
60
+ 382
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": "all_mpnet_base_v2_xnnpack_8da4w.pte",
79
  "precision": "8da4w",