all-MiniLM-L6-v2

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

Upstream model: all-MiniLM-L6-v2

Variants

XNNPACK int8

xnnpack/all_minilm_l6_v2_xnnpack_int8.pte is 8da8w (int8 dynamic activations, int8 per-channel weights) 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                                38 B
coreml/all_minilm_l6_v2_coreml_fp16.pte    43.4 MB
coreml/all_minilm_l6_v2_coreml_int8.pte    21.8 MB
coreml/config.json                         1.7 kB
tokenizer.json                             695 kB
tokenizer_config.json                      350 B
vulkan/all_minilm_l6_v2_vulkan_fp16.pte    43.1 MB
vulkan/config.json                         929 B
xnnpack/all_minilm_l6_v2_xnnpack_fp32.pte  86.2 MB
xnnpack/all_minilm_l6_v2_xnnpack_int8.pte  21.9 MB
xnnpack/config.json                        1.5 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.