fix typo

huggingface · Sep 20, 2024 · 617726c · 617726c
1 parent 89b56a4
commit 617726c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
@@ -1874,9 +1874,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()