Add specific numpy version

Zyphra · Nov 26, 2024 · 1b182f4 · 1b182f4
1 parent 6ec9a70
commit 1b182f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -153,6 +153,7 @@ def append_nvcc_threads(nvcc_extra_args):
         "einops",
         "transformers",
         "triton==2.2.0",
+        "numpy==1.26.4"
         # "mamba-ssm==2.1.0",
         # "causal-conv1d==1.3.0.post1",
     ],