Upgrade PyTorch to 2.1 (#1419)

mgoin · web-flow · commit d6d1d6b1dcc0 · 2023-11-20T07:10:35.000-07:00
diff --git a/setup.py b/setup.py
@@ -129,9 +129,9 @@ def _parse_requirements_file(file_path):
 _onnxruntime_deps = [
     "onnxruntime>=1.7.0",
 ]
-_torch_deps = ["torch>=1.7.0,<=2.0"]
+_torch_deps = ["torch>=1.7.0,<2.2"]
 _computer_vision_deps = [
-    "torchvision>=0.3.0,<0.16",
+    "torchvision>=0.3.0,<0.17",
     "opencv-python<=4.6.0.66",
 ]
 _openpifpaf_integration_deps = [

Original file line number	Diff line number	Diff line change
`@@ -129,9 +129,9 @@ def _parse_requirements_file(file_path):`
`129`	`129`	`_onnxruntime_deps = [`
`130`	`130`	`"onnxruntime>=1.7.0",`
`131`	`131`	`]`
`132`		`-_torch_deps = ["torch>=1.7.0,<=2.0"]`
	`132`	`+_torch_deps = ["torch>=1.7.0,<2.2"]`
`133`	`133`	`_computer_vision_deps = [`
`134`		`- "torchvision>=0.3.0,<0.16",`
	`134`	`+ "torchvision>=0.3.0,<0.17",`
`135`	`135`	`"opencv-python<=4.6.0.66",`
`136`	`136`	`]`
`137`	`137`	`_openpifpaf_integration_deps = [`