Bound transformers version in requirements (NVIDIA#7620)

Signed-off-by: Abhishree <[email protected]> Signed-off-by: Sasha Meister <[email protected]>
ssh-meister · Oct 5, 2023 · 33b0142 · 33b0142
1 parent 1a752e8
commit 33b0142
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