We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent df864aa commit b1b7f06Copy full SHA for b1b7f06
1 file changed
tests/models/t5gemma2/test_modeling_t5gemma2.py
@@ -621,6 +621,8 @@ def create_and_check_cross_attention_cache_is_not_sliding(
621
lm_labels,
622
pixel_values,
623
):
624
+ config.decoder.sliding_window = self.encoder_seq_length // 2
625
+ self.parent.assertGreater(self.encoder_seq_length, config.decoder.sliding_window)
626
model = self.causal_lm_class(config=config).to(torch_device).eval()
627
output = model.generate(
628
input_ids,
0 commit comments