Add SSM and Hybrid Models Test

Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
2025-03-15 17:26:01 +00:00
parent 61c6a5a796
commit 696245c2fc
2 changed files with 571 additions and 566 deletions
@@ -38,8 +38,6 @@ from .utils import (is_pp_missing_parameter,
                    make_empty_intermediate_tensors_factory, make_layers,
                    maybe_prefix)

-KVCache = Tuple[torch.Tensor, torch.Tensor]
-

 class BambaMLP(nn.Module):