model and files

convert.py

@@ -0,0 +1,24 @@
+#!pip install uv
+#!uv pip install transformers torch onnx
+# Импорты
+from transformers import AutoTokenizer, AutoModel
+import torch
+import onnx
+
+tokenizer = AutoTokenizer.from_pretrained("ai-forever/sbert_large_nlu_ru")
+model = AutoModel.from_pretrained("ai-forever/sbert_large_nlu_ru")
+
+# Экспорт в ONNX
+dummy_input = tokenizer("Тест", return_tensors="pt", padding=True, truncation=True)
+torch.onnx.export(model,
+                  (dummy_input['input_ids'], dummy_input['attention_mask']),
+                  "sbert_large_nlu_ru.onnx",
+                  input_names=["input_ids", "attention_mask"],
+                  output_names=["last_hidden_state"],
+                  dynamic_axes={
+                      "input_ids": {0: "batch", 1: "sequence"},
+                      "attention_mask": {0: "batch", 1: "sequence"},
+                      "last_hidden_state": {0: "batch", 1: "sequence", 2: "features"}
+                  })
+# from google.colab import files
+# files.download("/content/sbert_large_nlu_ru.onnx")

sbert_large_nlu_ru.onnx

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