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

Output attention takes an s (#6903)

Browse Source 
* Fix output_attention -> output_attentions

* Formatting

* One unsaved file
This commit is contained in:
Sylvain Gugger
2020-09-02 08:11:45 -04:00
committed by GitHub
parent 485da7222f
commit 8f2723caf0
11 changed files with 75 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modelcard.py
View File

@@ -122,7 +122,7 @@ class ModelCard:
modelcard = ModelCard.from_pretrained('bert-base-uncased') # Download model card from S3 and cache.
modelcard = ModelCard.from_pretrained('./test/saved_model/') # E.g. model card was saved using `save_pretrained('./test/saved_model/')`
modelcard = ModelCard.from_pretrained('./test/saved_model/modelcard.json')
modelcard = ModelCard.from_pretrained('bert-base-uncased', output_attention=True, foo=False)
modelcard = ModelCard.from_pretrained('bert-base-uncased', output_attentions=True, foo=False)
"""
cache_dir = kwargs.pop("cache_dir", None)