numpy 2.1.0 was released, but incompatible with numba (#1849) [skip ci]

axolotl-ai-cloud · Aug 22, 2024 · c3fc529 · c3fc529
1 parent 957c956
commit c3fc529
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -21,7 +21,7 @@ optimum==1.16.2
 hf_transfer
 colorama
 numba
-numpy>=1.24.4
+numpy>=1.24.4,<=2.0.1
 # qlora things
 evaluate==0.4.1
 scipy