Bound transformers version in requirements (NVIDIA#7620)

Signed-off-by: Abhishree <abhishreetm@gmail.com> Signed-off-by: Sasha Meister <sasha.meister.work@gmail.com>
ssh-meister · Oct 10, 2023 · 5f01aab · 5f01aab
1 parent 2da5c02
commit 5f01aab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt
@@ -2,6 +2,6 @@ hydra-core>1.3,<=1.3.2
 omegaconf<=2.3
 pytorch-lightning>=2.0,<=2.0.7
 torchmetrics>=0.11.0
-transformers>=4.0.1
+transformers>=4.0.1,<=4.33.3
 wandb
 webdataset>=0.1.48,<=0.1.62