From 60f5606c2da233062b166514e3df461de1ae3f5a Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lstein@gmail.com>
Date: Sat, 29 Jul 2023 13:28:29 -0400
Subject: [PATCH] downgrade torchmetrics to fix model import problem

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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",