Let's YOLO a serving output too

huggingface · May 9, 2023 · a98263b · a98263b
1 parent c7b8bb9
commit a98263b
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
@@ -1252,13 +1252,10 @@ def get_serving_input_signature(self) -> List[Dict[str, tf.TensorSpec]]:
 
     def serving_output(self, output):
         """
-        Prepare the output of the saved model. Each model must implement this function.
-
-        Args:
-            output ([`TFBaseModelOutput`]):
-                The output returned by the model.
+        Prepare the output of the saved model. Can be overridden if specific serving modifications are required.
         """
-        raise NotImplementedError
+
+        return output
 
     def can_generate(self) -> bool:
         """