relax versions for torch and torchvision

2023-10-22 20:32:29 -07:00 · 2023-10-22 20:32:29 -07:00 · 644b6ef415
parent aa6dff1f51
commit 644b6ef415
1 changed files with 2 additions and 2 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -14,8 +14,8 @@ classifiers = [
 ]

 dependencies = [
-    "torch==2.1.0",
-    "torchvision==0.16.0",
+    "torch",
+    "torchvision",
    "pillow==9.4.0",
    "opencv-python==4.8.1.78",
    "pyyaml==6.0.1",