jilijeanlouis commited on
Commit
f7af799
1 Parent(s): 41b5703

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

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