Skip to content

Commit

Permalink
fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
ArthurZucker committed Jul 23, 2024
1 parent 666c622 commit cc19ced
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/cache_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit cc19ced

Please sign in to comment.