From 52e5b6bd4aa9107de093e1be57ab12eeb3fbf36b Mon Sep 17 00:00:00 2001 From: Matt Date: Wed, 24 May 2023 16:05:47 +0100 Subject: [PATCH] output.keys() -> output --- src/transformers/modeling_tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index 38a4e8cc10512d..cd8e6956448778 100644 --- 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):