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

fix switch transformers

Browse Source
This commit is contained in:
raushan 2025-07-02 11:45:46 +02:00
parent f19d166d2b
commit ab7fac4a01
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/switch_transformers/modeling_switch_transformers.py
View File

@ -979,9 +979,7 @@ class SwitchTransformersStack(SwitchTransformersPreTrainedModel):
attention_mask,
inputs_embeds,
cache_position,
past_key_values.self_attention_cache
if isinstance(past_key_values, EncoderDecoderCache) is not None
else None,
past_key_values.self_attention_cache if isinstance(past_key_values, EncoderDecoderCache) else None,
output_attentions,
)
else: