Skip to content

Commit 6903c29

Browse files
authored
Merge branch 'master' into dcgan_fashiongen_example
2 parents f853048 + 411ab09 commit 6903c29

File tree

1 file changed

+5
-5
lines changed
  • examples/FasterTransformer_HuggingFace_Bert

1 file changed

+5
-5
lines changed

examples/FasterTransformer_HuggingFace_Bert/README.md

+5-5
Original file line numberDiff line numberDiff line change
@@ -80,13 +80,13 @@ cd /workspace/serve/examples/FasterTransformer_HuggingFace_Bert
8080

8181
# make sure to change the ../Huggingface_Transformers/setup_config.json "save_mode":"torchscript" and "FasterTransformer":true
8282

83-
# change the ../Huggingface_Transformers/setup_config.json
83+
# change the ../Huggingface_Transformers/setup_config.json as below
8484
{
8585
"model_name":"bert-base-uncased",
86-
"mode":"question_answering",
86+
"mode":"sequence_classification",
8787
"do_lower_case":true,
8888
"num_labels":"0",
89-
"save_mode":"pretrained",
89+
"save_mode":"torchscript",
9090
"max_length":"128",
9191
"captum_explanation":false,
9292
"embedding_name": "bert",
@@ -110,13 +110,13 @@ curl -X POST http://127.0.0.1:8080/predictions/my_tc -T ../Huggingface_Transform
110110
```bash
111111
# Question answering
112112

113-
# change the ../Huggingface_Transformers/setup_config.json
113+
# change the ../Huggingface_Transformers/setup_config.json as below
114114
{
115115
"model_name":"bert-base-uncased",
116116
"mode":"question_answering",
117117
"do_lower_case":true,
118118
"num_labels":"0",
119-
"save_mode":"pretrained",
119+
"save_mode":"torchscript",
120120
"max_length":"128",
121121
"captum_explanation":false,
122122
"embedding_name": "bert",

0 commit comments

Comments
 (0)