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

Add XNNPACK 8da4w export

Browse files
.gitattributes CHANGED
@@ -38,3 +38,4 @@ multi-qa-mpnet-base-dot-v1_xnnpack.pte filter=lfs diff=lfs merge=lfs -text
38
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.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
 
 
38
  xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.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
README.md CHANGED
@@ -18,19 +18,32 @@ Upstream model: [multi-qa-mpnet-base-dot-v1](https://huggingface.co/sentence-tra
18
  | --- | --- | --- |
19
  | `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte` | vulkan | fp16 |
20
  | `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte` | vulkan | int8 |
 
21
  | `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte` | xnnpack | fp32 |
22
 
 
 
 
 
 
 
 
 
 
 
 
23
  ## Repository structure
24
 
25
  ```
26
- config.json 48 B
27
- tokenizer.json 694 kB
28
- tokenizer_config.json 363 B
29
- vulkan/config.json 1.6 kB
30
- vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte 208 MB
31
- vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte 127 MB
32
- xnnpack/config.json 951 B
33
- xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte 416 MB
 
34
  ```
35
 
36
  ## Compatibility
 
18
  | --- | --- | --- |
19
  | `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte` | vulkan | fp16 |
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
36
 
37
  ```
38
+ config.json 48 B
39
+ tokenizer.json 694 kB
40
+ 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
xnnpack/config.json CHANGED
@@ -40,6 +40,39 @@
40
  ]
41
  }
42
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
43
  }
44
  ]
45
  }
 
40
  ]
41
  }
42
  }
43
+ },
44
+ {
45
+ "file": "multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte",
46
+ "precision": "8da4w",
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
  }
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4942c08d111e825df300d90ae780882d457721ab39f27d731031db97ee3bdf68
3
+ size 72474880