Cache: use batch_size instead of max_batch_size (#32657)

* more precise name * better docstrings * Update src/transformers/cache_utils.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
2024-08-16 11:48:45 +01:00
parent 8f9fa3b081
commit cf32ee1753
9 changed files with 112 additions and 54 deletions
--- a/utils/check_docstrings.py
+++ b/utils/check_docstrings.py
@@ -75,6 +75,11 @@ OBJECTS_TO_IGNORE = [
    "TFSequenceSummary",
    "TFBertTokenizer",
    "TFGPT2Tokenizer",
+    # Going through an argument deprecation cycle, remove after v4.46
+    "HybridCache",
+    "MambaCache",
+    "SlidingWindowCache",
+    "StaticCache",
    # Missing arguments in the docstring
    "ASTFeatureExtractor",
    "AlbertModel",