diff --git a/pyproject.toml b/pyproject.toml index eabc445c1a..9cad874ccf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -81,7 +81,7 @@ dependencies = [ "test-tube~=0.7.5", "torch~=2.0.1", "torchvision~=0.15.2", - "torchmetrics~=1.0.1", + "torchmetrics~=0.11.0", "torchsde~=0.2.5", "transformers~=4.31.0", "uvicorn[standard]~=0.21.1",