fix numpy<2.0 to address numerous bugs

Lightning-Universe · Jul 5, 2024 · ef18d5c · ef18d5c
1 parent e333fd8
commit ef18d5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements_litpose.txt b/requirements_litpose.txt
@@ -7,6 +7,7 @@ kornia==0.7.2
 lightning
 matplotlib==3.9.0
 moviepy==1.0.3
+numpy<2.0.0
 nvidia-dali-cuda110  # freeze this to be cuda 11 for now
 opencv-python-headless==4.8.1.78
 pandas==2.2.2
@@ -21,4 +22,4 @@ torchtyping==0.1.4
 torchvision==0.18.1
 typeguard==4.3.0
 typing==3.7.4.3
-optax==0.1.7
+optax==0.1.7