diff --git a/requirements.txt b/requirements.txt index 9ff1955f..16ca7fc4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ pytorch-lightning>=1.8.0, <1.9.5 Pillow>=9.0, <10.1.1 tabulate>=0.9.0, <0.9.1 tensorboardX>=2.5.0, <2.6.3 -torchmetrics>=0.11.0, <0.11.5 +torchmetrics>=0.11.0, <1.4.3 torchvision>=0.13.0, <0.16.2 deepspeed>=0.9.0, <0.12.5 datasets>=2.9.0, <2.15.1