Read __class__.__name__ properly

huggingface · May 9, 2023 · cf61e97 · cf61e97
1 parent a98263b
commit cf61e97
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
@@ -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