action

Author: xadupre

.github/workflows/documentation.yml

@@ -24,7 +24,8 @@ jobs:
         with:
           python-version: ${{ matrix.python }}
 
-      - uses: tlylt/install-graphviz@v1
+      - name: install graphviz
+        uses: ts-graphviz/setup-graphviz@v2
 
       - name: Install pandoc
         run: sudo apt-get install -y pandoc

_doc/conf.py

@@ -119,6 +119,7 @@ def linkcode_resolve(domain, info):
     ("py:class", "False"),
     ("py:class", "True"),
     ("py:class", "Argument"),
+    ("py:class", "CacheProcessor"),
     ("py:class", "default=sklearn.utils.metadata_routing.UNCHANGED"),
     ("py:class", "diffusers.models.unets.unet_2d_condition.UNet2DConditionOutput"),
     ("py:class", "ModelProto"),
@@ -142,6 +143,7 @@ def linkcode_resolve(domain, info):
     ("py:class", "transformers.cache_utils.EncoderDecoderCache"),
     ("py:class", "transformers.cache_utils.HybridCache"),
     ("py:class", "transformers.cache_utils.MambaCache"),
+    ("py:class", "transformers.models.mamba.modeling_mamba.MambaCache"),
     ("py:class", "transformers.cache_utils.SlidingWindowCache"),
     ("py:class", "transformers.cache_utils.StaticCache"),
     ("py:class", "transformers.configuration_utils.PretrainedConfig"),