riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[Attention] Small fix on output attentions (#38948)

Browse Source 
small fix
This commit is contained in:
Anton Vlasjuk 2025-06-24 14:42:10 +02:00 committed by GitHub
parent 4f650040a6
commit 23c89a6732
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/configuration_utils.py
View File

@ -338,7 +338,7 @@ class PretrainedConfig(PushToHubMixin):
@output_attentions.setter @output_attentions.setter
def output_attentions(self, value): def output_attentions(self, value):
if self._attn_implementation != "eager": if value is True and self._attn_implementation != "eager":
raise ValueError( raise ValueError(
"The `output_attentions` attribute is not supported when using the `attn_implementation` set to " "The `output_attentions` attribute is not supported when using the `attn_implementation` set to "
f"{self._attn_implementation}. Please set it to 'eager' instead." f"{self._attn_implementation}. Please set it to 'eager' instead."