From 898aef5158cbbfcc205d13dff4a07ccd9fcfdcbd Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Wed, 15 Mar 2023 10:11:27 +0100 Subject: [PATCH] ci: add mapping for PT 2.0 (#1622) --- requirements/adjust-versions.py | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements/adjust-versions.py b/requirements/adjust-versions.py index e24ffbd92ed..ecfbe3755b5 100644 --- a/requirements/adjust-versions.py +++ b/requirements/adjust-versions.py @@ -20,6 +20,7 @@ from packaging.version import Version VERSIONS = [ + {"torch": "2.0.0", "torchvision": "0.15.0", "torchtext": "0.15.0"}, # stable {"torch": "1.14.0", "torchvision": "0.15.0", "torchtext": "0.15.0"}, # nightly {"torch": "1.13.1", "torchvision": "0.14.1", "torchtext": "0.14.1"}, # stable {"torch": "1.13.0", "torchvision": "0.14.0", "torchtext": "0.14.0"},