downgrade torchmetrics to fix model import problem

2026-04-23 03:00:31 -04:00 · 2023-07-29 13:28:29 -04:00
parent 71768f5988
commit 60f5606c2d
1 changed files with 1 additions and 1 deletions
--- 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",