initial commit

.gitattributes

@@ -0,0 +1 @@
+*.onnx filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.venv

README.md

@@ -0,0 +1,2 @@
+# ONNX port of sentence-transormers/all-MiniLM-L6-v2
+

convert.py

@@ -0,0 +1,11 @@
+from transformers import AutoTokenizer, AutoModel
+import torch
+
+model = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+model.eval()
+
+sample = torch.randint(low=0, high=1, size=(1,128))
+input = (sample, sample, sample)
+
+torch.onnx.export(model, input, 'pytorch_model.onnx', export_params=True)
+

pytorch_model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ec02f3d61fd002cc3c4220ba2f7851fad67abf4004312753f866a9c37a5693c
+size 90933591

requirements.txt

@@ -0,0 +1,3 @@
+sentence-transformers==2.2.2
+torch==2.0.0
+onnx=1.13.1