Skip to content

Commit 27c3807

Browse files
authored
[Pop2Piano] Fix cache usage (#42170)
fix
1 parent ffb35fe commit 27c3807

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/transformers/models/pop2piano/modeling_pop2piano.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -750,7 +750,8 @@ def forward(
750750
position_bias,
751751
encoder_hidden_states,
752752
encoder_extended_attention_mask,
753-
encoder_decoder_position_bias, # as a positional argument for gradient checkpointing past_key_values=past_key_values,
753+
encoder_decoder_position_bias, # as a positional argument for gradient checkpointing
754+
past_key_values=past_key_values,
754755
use_cache=use_cache,
755756
output_attentions=output_attentions,
756757
cache_position=cache_position,

0 commit comments

Comments
 (0)