diff --git a/pyproject.toml b/pyproject.toml
index 5978fa2..976a91f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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",