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

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:

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.