c-my/transformers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Better warning

Browse Source
This commit is contained in:
Lysandre
2020-06-09 19:08:27 -04:00
parent 6342e803a9
commit 513ba3b6a0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/transformers/configuration_utils.py
+2 -2
View File
@@ -59,7 +59,7 @@ class PretrainedConfig(object):
if "output_attentions" in kwargs:
warnings.warn(
"The `output_attentions` in the configuration is deprecated and will be removed in a later version. "
"Please use `output_attention` as an argument to your model instead.",
"Please use `output_attention` as an argument to your model's foward method instead.",
FutureWarning,
)
self.output_attentions = kwargs.pop("output_attentions", False)
@@ -298,7 +298,7 @@ class PretrainedConfig(object):
if "output_attentions" in kwargs:
warnings.warn(
"The `output_attentions` in the configuration is deprecated and will be removed in a later version. "
"Please use `output_attention` as an argument to your model instead.",
"Please use `output_attention` as an argument to your model's forward method instead.",
FutureWarning,
)