fix(general_perf): fix deberta inputs mismatch prob

hliuca · Apr 8, 2024 · 7b63c32 · 7b63c32
1 parent 3662456
commit 7b63c32
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