From 94faa5de148ce5d515c440f3c9a4901b12893d6e Mon Sep 17 00:00:00 2001
From: Zadagu <Zadagu@users.noreply.github.com>
Date: Thu, 6 Jul 2023 16:16:02 +0200
Subject: [PATCH] fixes ImportError described in #3658.

The issue was introduced by a new release of torchmetrics.
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 8374471a0d..a764bda109 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -82,7 +82,7 @@ dependencies = [
   "test-tube>=0.7.5",
   "torch~=2.0.0",
   "torchvision>=0.14.1",
-  "torchmetrics",
+  "torchmetrics==0.11.4",
   "torchsde==0.2.5",
   "transformers~=4.30",
   "uvicorn[standard]==0.21.1",