riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 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, attention_mask,
inputs_embeds, inputs_embeds,
cache_position, cache_position,
past_key_values.self_attention_cache past_key_values.self_attention_cache if isinstance(past_key_values, EncoderDecoderCache) else None,
if isinstance(past_key_values, EncoderDecoderCache) is not None
else None,
output_attentions, output_attentions,
) )
else: else: