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 index 083d1181..1163161b 100644 --- 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