diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index dc1af32f3b36..462cdd664d79 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -613,8 +613,8 @@ def docstring_decorator(fn): PT_RETURN_INTRODUCTION = r""" Returns: - :class:`~{full_output_type}` or :obj:`tuple(torch.FloatTensor)`: A :class:`~{full_output_type}` (if - ``return_dict=True`` is passed or when ``config.return_dict=True``) or a tuple of :obj:`torch.FloatTensor` + :class:`~{full_output_type}` or :obj:`tuple(torch.FloatTensor)`: A :class:`~{full_output_type}` or + a tuple of :obj:`torch.FloatTensor` (if ``return_dict=False`` is passed or when ``config.return_dict=False``) comprising various elements depending on the configuration (:class:`~transformers.{config_class}`) and inputs. """ @@ -622,9 +622,9 @@ def docstring_decorator(fn): TF_RETURN_INTRODUCTION = r""" Returns: - :class:`~{full_output_type}` or :obj:`tuple(tf.Tensor)`: A :class:`~{full_output_type}` (if - ``return_dict=True`` is passed or when ``config.return_dict=True``) or a tuple of :obj:`tf.Tensor` comprising - various elements depending on the configuration (:class:`~transformers.{config_class}`) and inputs. + :class:`~{full_output_type}` or :obj:`tuple(tf.Tensor)`: A :class:`~{full_output_type}` or + a tuple of :obj:`tf.Tensor` (if ``return_dict=False`` is passed or when ``config.return_dict=False``) + comprising various elements depending on the configuration (:class:`~transformers.{config_class}`) and inputs. """