diff --git a/pyproject.toml b/pyproject.toml
index d33cf8f..f353222 100644
--- 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",