fix batch size

Author: xadupre

.github/workflows/ci.yml

@@ -16,7 +16,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python: ['3.11', '3.12']
-        transformers: ['4.48.3', '4.50.3', 'main']
+        transformers: ['4.48.3', '4.51.1', 'main']
         torch: ['2.6', 'main']
 
     steps:

_doc/recipes/plot_dynamic_shapes_max.py

@@ -78,7 +78,7 @@ def forward(self, x, y, fact):
 def make_undefined_dimension(i: int) -> torch.SymInt:
     """
     Uses for a custom op when a new dimension must be introduced to bypass
-    some verficiation. The following function creates a dummy output
+    some verification. The following function creates a dummy output
     with a dimension based on the content.
 
     .. code-block:: python

_unittests/ut_torch_export_patches/test_onnx_export_errors.py

@@ -114,7 +114,7 @@ def forward(self, x: torch.Tensor, cache: MambaCache):
         DYN = torch.export.Dim.DYNAMIC
 
         with bypass_export_some_errors():
-            cache = MambaCache(_config(), max_batch_size=1, device="cpu")
+            cache = MambaCache(_config(), max_batch_size=2, device="cpu")
             torch.export.export(
                 Model(),
                 (x, cache),