| name: "bhadresh-savani_distilbert-base-uncased-emotion_onnx_model" | |
| max_batch_size: 0 | |
| platform: "onnxruntime_onnx" | |
| default_model_filename: "model.bin" | |
| input [ | |
| { | |
| name: "input_ids" | |
| data_type: TYPE_INT32 | |
| dims: [-1, -1] | |
| }, | |
| { | |
| name: "attention_mask" | |
| data_type: TYPE_INT32 | |
| dims: [-1, -1] | |
| } | |
| ] | |
| output { | |
| name: "output" | |
| data_type: TYPE_FP32 | |
| dims: [-1, 6] | |
| } | |
| instance_group [ | |
| { | |
| count: 1 | |
| kind: KIND_GPU | |
| } | |
| ] |