Merge branch 'master' into bert-readme-update

LaurinmyReha · Nov 3, 2021 · 6ed2f95 · 6ed2f95
2 parents 3956ac2 + 9a453b4
commit 6ed2f95
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/Huggingface_Transformers/Transformer_handler_generalized.py b/examples/Huggingface_Transformers/Transformer_handler_generalized.py
@@ -183,7 +183,7 @@ def inference(self, input_batch):
             # the output should be only answer_start and answer_end
             # we are outputing the words just for demonstration.
             if self.setup_config["save_mode"]=="pretrained":
-                outputs = self.model(input_batch)
+                outputs = self.model(input_ids_batch,attention_mask_batch)
                 answer_start_scores = outputs.start_logits
                 answer_end_scores = outputs.end_logits
             else:

diff --git a/frontend/server/src/main/java/org/pytorch/serve/util/ApiUtils.java b/frontend/server/src/main/java/org/pytorch/serve/util/ApiUtils.java
@@ -402,7 +402,7 @@ public static RestJob addRESTInferenceJob(
     public static String getInferenceErrorResponseMessage(String modelName, String modelVersion) {
         String responseMessage = "Model \"" + modelName;
 
-        if (modelVersion == null) {
+        if (modelVersion != null) {
             responseMessage += "\" Version " + modelVersion;
         }
 

diff --git a/requirements/developer.txt b/requirements/developer.txt
@@ -9,5 +9,5 @@ nvgpu==0.8.0; sys_platform == 'win32'
 grpcio
 protobuf
 grpcio-tools
-transformers==4.6.0
+transformers==4.11.0
 pygit2