Oops

2026-01-28 14:07:19 +00:00
parent 1e1eb8f8a1
commit 07c4be157b
1 changed files with 2 additions and 2 deletions
@@ -681,8 +681,8 @@ class MiniMaxShardingStrategy(TensorParallelShardingStrategy):
        def __call__(
            self,
            x: mx.array,
-            mask: Optional[mx.array] = None,
-            cache: Optional[Any] = None,
+            mask: mx.array | Any = None,
+            cache: Any | None = None,
        ) -> mx.array:
            B, L, D = x.shape