From 644b6ef415f267d9c0466e691cff4d2adb760533 Mon Sep 17 00:00:00 2001
From: Jianwei Yang <jwyang@users.noreply.github.com>
Date: Sun, 22 Oct 2023 20:32:29 -0700
Subject: [PATCH] relax versions for torch and torchvision

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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",