downgrade torchmetrics to fix model import problem

This commit is contained in:
Lincoln Stein 2023-07-29 13:28:29 -04:00
parent 71768f5988
commit 60f5606c2d

View File

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