Update requirements-pytorch.txt

awslabs · Oct 18, 2023 · 8f2163d · 8f2163d
1 parent 257bfc9
commit 8f2163d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt
@@ -1,7 +1,7 @@
 torch>=1.9,<3
-pytorch_lightning>=1.8,<2.2
+pytorch-lightning>=1.5,<3
 # Need to pin protobuf (for now)
 # See: https://github.com/PyTorchLightning/pytorch-lightning/issues/13159
 protobuf~=3.19.0
 scipy~=1.10; python_version > "3.7.0"
-scipy~=1.7.3; python_version <= "3.7.0"
+scipy~=1.7.3; python_version <= "3.7.0"