fix some of the edge cases for Jamba (#1452)

* fix some of the edge cases for Jamba * update requirements for jamba
2024-03-29 02:38:02 -04:00
parent e634118f90
commit 05b398a072
8 changed files with 92 additions and 17 deletions
--- a/setup.py
+++ b/setup.py
@@ -78,7 +78,7 @@ setup(
            "deepspeed-kernels",
        ],
        "mamba-ssm": [
-            "mamba-ssm==1.0.1",
+            "mamba-ssm==1.2.0.post1",
        ],
        "auto-gptq": [
            "auto-gptq==0.5.1",