limit torch to <=2.1 to avoid type errors (#2857)

This commit is contained in:
Justin Kerr authored and GitHub committed 2024-01-31 19:34:00 +00:00
1 parent 3a90cb529f
commit 17748a8309
1 file changed
+2 -1
+2 -1
View File
@@ -53,7 +53,8 @@ dependencies = [
"scikit-image>=0.19.3",
"splines==0.3.0",
"tensorboard>=2.13.0",
"torch>=1.13.1",
# pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
"torch>=1.13.1,<=2.1",
"torchvision>=0.14.1",
"torchmetrics[image]>=1.0.1",
"typing_extensions>=4.4.0",