From 9eb267b7419152b93a284c735cd9796c3b848695 Mon Sep 17 00:00:00 2001
From: Sean Naren <snarenthiran@nvidia.com>
Date: Thu, 1 Dec 2022 21:12:44 +0000
Subject: [PATCH] Upperbound the torchmetrics version (#5537)

Signed-off-by: andrusenkoau <andrusenkoau@gmail.com>
---
 requirements/requirements_lightning.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt
index dcde2c6bd810..9876002b806d 100644
--- a/requirements/requirements_lightning.txt
+++ b/requirements/requirements_lightning.txt
@@ -2,7 +2,7 @@ hydra-core>=1.2.0,<1.3
 omegaconf>=2.2,<2.3
 pytorch-lightning>=1.8.3
 pyyaml<6  # Pinned until omegaconf works with pyyaml>=6
-torchmetrics>=0.4.1rc0
+torchmetrics>=0.4.1rc0,<=0.10.3
 transformers>=4.0.1,<=4.21.2
 wandb
 webdataset>=0.1.48,<=0.1.62