From 014e453428f234ab84d1a3aa72998d8817e27f70 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Fri, 20 Sep 2024 17:09:35 +0200 Subject: [PATCH] fix typo --- tests/generation/test_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py index 7b7226ca3e62e9..3af5cbebf7024b 100644 --- a/tests/generation/test_utils.py +++ b/tests/generation/test_utils.py @@ -1846,9 +1846,9 @@ def test_new_cache_format(self, num_beams, do_sample): if config.is_encoder_decoder: cache_cls = EncoderDecoderCache past_key_values = cache_cls(DynamicCache(), DynamicCache()) - elif getattr(self.config, "sliding_window", None) is not None: + elif getattr(config, "sliding_window", None) is not None: cache_cls = DynamicSlidingWindowCache - past_key_values = cache_cls(self.config.sliding_window) + past_key_values = cache_cls(config.sliding_window) else: cache_cls = DynamicCache past_key_values = cache_cls()