From cf61e978a1b7f41b4c7a327f9435afba3e3072b8 Mon Sep 17 00:00:00 2001 From: Matt Date: Tue, 9 May 2023 17:00:23 +0100 Subject: [PATCH] Read __class__.__name__ properly --- 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 70798aa0977188..f68f073f6a9f43 100644 --- a/src/transformers/modeling_tf_utils.py +++ b/src/transformers/modeling_tf_utils.py @@ -1230,7 +1230,7 @@ def eager_serving(self, inputs): def get_serving_input_signature(self) -> List[Dict[str, tf.TensorSpec]]: model_inputs = list(dict(inspect.signature(self.call).parameters).keys()) sig = {} - if self.__name__.endswith("ForMultipleChoice"): + if self.__class__.__name__.endswith("ForMultipleChoice"): text_dims = 3 else: text_dims = 2