InterDigitalInc
diff --git a/‎pyproject.toml‎
Lines changed: 2 additions & 0 deletions b/‎pyproject.toml‎
Lines changed: 2 additions & 0 deletions
@@ -36,6 +36,8 @@ dependencies = [
     "pillow>=9.0,<10.0",
     "pip>=25.0.1",
     "ptflops>=0.7.3",
+    "torch>=2.0.0",
+    "torchvision>=0.15.1",
     "wheel>=0.45.1",
     "yuvio",
 ]
Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,8 @@ dependencies = [`
`36`	`36`	`"pillow>=9.0,<10.0",`
`37`	`37`	`"pip>=25.0.1",`
`38`	`38`	`"ptflops>=0.7.3",`
	`39`	`+ "torch>=2.0.0",`
	`40`	`+ "torchvision>=0.15.1",`
`39`	`41`	`"wheel>=0.45.1",`
`40`	`42`	`"yuvio",`
`41`	`43`	`]`