Skip to content

Commit b1b7f06

Browse files
committed
upd test
1 parent df864aa commit b1b7f06

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/models/t5gemma2/test_modeling_t5gemma2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -621,6 +621,8 @@ def create_and_check_cross_attention_cache_is_not_sliding(
621621
lm_labels,
622622
pixel_values,
623623
):
624+
config.decoder.sliding_window = self.encoder_seq_length // 2
625+
self.parent.assertGreater(self.encoder_seq_length, config.decoder.sliding_window)
624626
model = self.causal_lm_class(config=config).to(torch_device).eval()
625627
output = model.generate(
626628
input_ids,

0 commit comments

Comments
 (0)