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

RWKV: fix mask warning typo (#37114)

Browse Source 
rwkv: fix mask warning typo
This commit is contained in:
Robin Kahlow 2025-03-31 10:07:51 +01:00 committed by GitHub
parent 19919689b2
commit d1efaf0318
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/models/rwkv/modeling_rwkv.py
View File

@ -625,7 +625,7 @@ class RwkvModel(RwkvPreTrainedModel):
use_cache = use_cache if use_cache is not None else (self.config.use_cache if not self.training else False)
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
if attention_mask is None:
if attention_mask is not None:
logger.warning_once("`attention_mask` was passed, but it is unused in this model.")
if self.training == self.layers_are_rescaled: