make fixup

huggingface · May 10, 2023 · 6bc65d1 · 6bc65d1
1 parent 9c573aa
commit 6bc65d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
@@ -1256,12 +1256,12 @@ def serving_output(self, output):
         Prepare the output of the saved model. Can be overridden if specific serving modifications are required.
         """
         if isinstance(output, ModelOutput):
-            if 'hidden_states' in output:
+            if "hidden_states" in output:
                 try:
                     output.hidden_states = tf.convert_to_tensor(output.hidden_states)
                 except ValueError:
                     pass  # Layers may not have the same dimensions
-            if 'attentions' in output:
+            if "attentions" in output:
                 try:
                     output.attentions = tf.convert_to_tensor(output.attentions)
                 except ValueError: