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

file_utils.py: TF examples outputs.last_hidden_states -> state (#9382)

Browse Source
This commit is contained in:
Chris Kennedy
2021-01-02 11:58:16 -05:00
committed by GitHub
parent b01f451ca3
commit c4fd609afb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/file_utils.py
View File

@@ -851,7 +851,7 @@ TF_BASE_MODEL_SAMPLE = r"""
>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="tf") >>> inputs = tokenizer("Hello, my dog is cute", return_tensors="tf")
>>> outputs = model(inputs) >>> outputs = model(inputs)
>>> last_hidden_states = outputs.last_hidden_states >>> last_hidden_states = outputs.last_hidden_state
""" """
TF_MULTIPLE_CHOICE_SAMPLE = r""" TF_MULTIPLE_CHOICE_SAMPLE = r"""