|
name: "bhadresh-savani_distilbert-base-uncased-emotion_onnx_inference" |
|
max_batch_size: 0 |
|
platform: "ensemble" |
|
|
|
input [ |
|
{ |
|
name: "TEXT" |
|
data_type: TYPE_STRING |
|
dims: [ -1 ] |
|
} |
|
] |
|
|
|
output { |
|
name: "output" |
|
data_type: TYPE_FP32 |
|
dims: [-1, 6] |
|
} |
|
|
|
ensemble_scheduling { |
|
step [ |
|
{ |
|
model_name: "bhadresh-savani_distilbert-base-uncased-emotion_onnx_tokenize" |
|
model_version: -1 |
|
input_map { |
|
key: "TEXT" |
|
value: "TEXT" |
|
} |
|
output_map [ |
|
{ |
|
key: "input_ids" |
|
value: "input_ids" |
|
}, |
|
{ |
|
key: "attention_mask" |
|
value: "attention_mask" |
|
} |
|
] |
|
}, |
|
{ |
|
model_name: "bhadresh-savani_distilbert-base-uncased-emotion_onnx_model" |
|
model_version: -1 |
|
input_map [ |
|
{ |
|
key: "input_ids" |
|
value: "input_ids" |
|
}, |
|
{ |
|
key: "attention_mask" |
|
value: "attention_mask" |
|
} |
|
] |
|
output_map { |
|
key: "output" |
|
value: "output" |
|
} |
|
} |
|
] |
|
} |