Add: Preparation for ONNX inference support ④

2024-09-17 13:25:17 +09:00
parent ebd4551d1e
commit 9de6e04ad7
5 changed files with 272 additions and 30 deletions
--- a/bert/bert_models.json
+++ b/bert/bert_models.json
@@ -3,6 +3,10 @@
        "repo_id": "ku-nlp/deberta-v2-large-japanese-char-wwm",
        "files": ["pytorch_model.bin"]
    },
+    "deberta-v2-large-japanese-char-wwm-onnx": {
+        "repo_id": "tsukumijima/deberta-v2-large-japanese-char-wwm-onnx",
+        "files": ["model.onnx"]
+    },
    "chinese-roberta-wwm-ext-large": {
        "repo_id": "hfl/chinese-roberta-wwm-ext-large",
        "files": ["pytorch_model.bin"]