Merge pull request bytedance#63 from bytedance/fix_general_perf_deberta

fix(general_perf): fix deberta inputs mismatch prob
hliuca · Apr 10, 2024 · 63b599c · 63b599c
2 parents 3662456 + 7b63c32
commit 63b599c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/byte_infer_perf/general_perf/model_zoo/deberta-torch-fp32.json b/byte_infer_perf/general_perf/model_zoo/deberta-torch-fp32.json
@@ -5,10 +5,10 @@
     "framework_version": "1.10.0",
     "model_format": "pt",
     "model_precision": "FP32",
-    "inputs":"input_ids.1,attention_mask.1",
+    "inputs":"input_ids.1,attention_mask.1,token_type_ids.1",
     "outputs":"start_logits,end_logits",
-    "input_shape": {"input_ids.1": [1,384], "attention_mask.1": [1,384]},
-    "input_type": "LONG,LONG",
+    "input_shape": {"input_ids.1": [1,384], "attention_mask.1": [1,384], "token_type_ids.1": [1,384]},
+    "input_type": "LONG,LONG,LONG",
     "dataset_name": "open_squad",
     "max_batch_size": 64,
     "is_quantized": false