jilijeanlouis commited on
Commit
9761b7b
1 Parent(s): 298d942

Upload classification_philschmid_MiniLM-L6-H384-uncased-sst2_tensorrt_model/config.pbtxt with huggingface_hub

Browse files
classification_philschmid_MiniLM-L6-H384-uncased-sst2_tensorrt_model/config.pbtxt ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: "classification_philschmid_MiniLM-L6-H384-uncased-sst2_tensorrt_model"
2
+ max_batch_size: 0
3
+ platform: "tensorrt_plan"
4
+ default_model_filename: "model.bin"
5
+
6
+ input [
7
+ {
8
+ name: "input_ids"
9
+ data_type: TYPE_INT32
10
+ dims: [-1, -1]
11
+ },
12
+ {
13
+ name: "token_type_ids"
14
+ data_type: TYPE_INT32
15
+ dims: [-1, -1]
16
+ },
17
+ {
18
+ name: "attention_mask"
19
+ data_type: TYPE_INT32
20
+ dims: [-1, -1]
21
+ }
22
+ ]
23
+
24
+ output {
25
+ name: "output"
26
+ data_type: TYPE_FP32
27
+ dims: [-1, 2]
28
+ }
29
+
30
+ instance_group [
31
+ {
32
+ count: 1
33
+ kind: KIND_GPU
34
+ }
35
+ ]