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)