Add XNNPACK int8 (8da8w) variant
Browse files- .gitattributes +1 -0
- README.md +8 -4
- xnnpack/all_mpnet_base_v2_xnnpack_int8.pte +3 -0
- xnnpack/config.json +33 -0
.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/
|
| 27 |
-
|
| 28 |
-
|
|
|
|
| 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/
|
|
|
|
| 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",
|