all-mpnet-base-v2

This repository hosts the all-mpnet-base-v2 models exported for the React Native ExecuTorch library as ExecuTorch .pte programs, ready to run on device.

Upstream model: all-mpnet-base-v2

Variants

Path Backend Precision
vulkan/all_mpnet_base_v2_vulkan_fp16.pte vulkan fp16
vulkan/all_mpnet_base_v2_vulkan_int8.pte vulkan int8
xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte xnnpack 8da4w
xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte xnnpack fp32
xnnpack/all_mpnet_base_v2_xnnpack_int8.pte xnnpack int8

XNNPACK int8

xnnpack/all_mpnet_base_v2_xnnpack_int8.pte is 8da8w (int8 dynamic activations, int8 per-channel weights) and xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte is 8da4w (int4 weights, group 32), both on the Linear layers with an int8 word-embedding table. SQuAD v1.1 dev passage retrieval (2000 questions against their 315 paragraphs), dot product:

top-1 MRR
fp32 65.05 0.7631
int8 65.00 0.7633
8da4w 64.60 0.7606

Repository structure

config.json                                  39 B
tokenizer.json                               694 kB
tokenizer_config.json                        363 B
vulkan/all_mpnet_base_v2_vulkan_fp16.pte     208 MB
vulkan/all_mpnet_base_v2_vulkan_int8.pte     127 MB
vulkan/config.json                           1.5 kB
xnnpack/all_mpnet_base_v2_xnnpack_8da4w.pte  69.1 MB
xnnpack/all_mpnet_base_v2_xnnpack_fp32.pte   416 MB
xnnpack/all_mpnet_base_v2_xnnpack_int8.pte   105 MB
xnnpack/config.json                          2.2 kB

Compatibility

These files are published for the ExecuTorch v1.4.1 runtime. ExecuTorch gives no forward compatibility guarantee, so an older runtime may fail to load them.

To use them in React Native ExecuTorch, pass the model constant shipped in the library's model registry to the corresponding task pipeline. See the documentation.

To load these files in your own ExecuTorch runtime, read the compatibility note first.

Downloads last month
640
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Collection including software-mansion/react-native-executorch-all-mpnet-base-v2