From 6bc65d186a58cfae1035b675a99170d170b627e3 Mon Sep 17 00:00:00 2001
From: Matt <rocketknight1@gmail.com>
Date: Wed, 10 May 2023 15:27:07 +0100
Subject: [PATCH] make fixup

---
 src/transformers/modeling_tf_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
index f39c2783daef46..cce7536e2a0e87 100644
--- 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: