SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Correct indentation in docstring

Browse Source 
For some reason Sphinx extremely dislikes this and crashes.
This commit is contained in:
LysandreJik
2020-03-27 09:28:21 -04:00
parent 3ee431dd4c
commit e2c05f06ef
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/modeling_tf_utils.py
View File

@@ -231,7 +231,7 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
def save_pretrained(self, save_directory): def save_pretrained(self, save_directory):
""" Save a model and its configuration file to a directory, so that it """ Save a model and its configuration file to a directory, so that it
can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pretrained`` class method. can be re-loaded using the :func:`~transformers.PreTrainedModel.from_pretrained` class method.
""" """
assert os.path.isdir( assert os.path.isdir(
save_directory save_directory
@@ -541,7 +541,7 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
``1`` for tokens that are NOT MASKED, ``0`` for MASKED tokens. ``1`` for tokens that are NOT MASKED, ``0`` for MASKED tokens.
Defaults to `None`. Defaults to `None`.
`What are attention masks? <../glossary.html#attention-mask>`__ `What are attention masks? <../glossary.html#attention-mask>`__
decoder_start_token_id=None: (`optional`) int decoder_start_token_id=None: (`optional`) int
If an encoder-decoder model starts decoding with a different token than BOS. If an encoder-decoder model starts decoding with a different token than BOS.