Upgrade pytorch lightning version in requirements

Signed-off-by: Abhishree <abhishreetm@gmail.com>
NVIDIA · Apr 16, 2023 · 2b91e1b · 2b91e1b
1 parent 5c52406
commit 2b91e1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt
@@ -1,7 +1,7 @@
 hydra-core>=1.2.0,<1.3
 omegaconf>=2.2,<2.3
-pytorch-lightning>=1.9.0,<=1.9.4
-pyyaml<6  # Pinned until omegaconf works with pyyaml>=6
+pytorch-lightning>=2.0
+ipyyaml<6  # Pinned until omegaconf works with pyyaml>=6
 torchmetrics>=0.11.0
 transformers>=4.0.1
 wandb