diff --git a/pyproject.toml b/pyproject.toml index dd8d94b0..d2f13bc0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -84,7 +84,7 @@ classifiers = [ dependencies = [ "torch==2.7.0", "torchvision==0.22.0", - "torchmetrics[image]", + "torchmetrics[image]==1.7.4", "requests>=2.31.0", "transformers", "pytorch-lightning",