output.keys() -> output

huggingface · May 24, 2023 · 52e5b6b · 52e5b6b
1 parent 05f7584
commit 52e5b6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
@@ -1279,7 +1279,7 @@ def serving_output(self, output):
         """
         if not isinstance(output, ModelOutput):
             return output
-        for key in output.keys():
+        for key in output:
             if key.endswith("hidden_states") and not getattr(self.config, "output_hidden_states", False):
                 output[key] = None
             elif key.endswith("attentions") and not getattr(self.config, "output_attentions", False):