relax torch + torchvision version requirements

2025-03-23 14:27:57 -05:00
parent d33e01ee13
commit a6ba138800
1 changed files with 2 additions and 2 deletions
@@ -1,8 +1,8 @@
 [project]
 authors = [{ name = "Jianyuan Wang", email = "jianyuan@robots.ox.ac.uk" }]
 dependencies = [
-    "torch==2.3.1",
-    "torchvision==0.18.1",
+    "torch>=2.3.1",
+    "torchvision>=0.18.1",
    "numpy==1.26.1",
    "Pillow",
    "huggingface_hub",