From cc19ced8c8c462d8484ee83668c2275617b78ebb Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Tue, 23 Jul 2024 17:00:13 +0200 Subject: [PATCH] fixup --- src/transformers/cache_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/cache_utils.py b/src/transformers/cache_utils.py index 472beaccba1a90..1cfdbc7681c754 100644 --- a/src/transformers/cache_utils.py +++ b/src/transformers/cache_utils.py @@ -303,7 +303,7 @@ class DynamicCache(Cache): """ def __init__(self) -> None: - super().__init__(self) + super().__init__() self.key_cache: List[torch.Tensor] = [] self.value_cache: List[torch.Tensor] = [] self._seen_tokens = 0 # Used in `generate` to keep tally of how many tokens the cache has seen