ci: add mapping for PT 2.0 (#1622)

Lightning-AI · Mar 15, 2023 · 898aef5 · 898aef5
1 parent c8d649d
commit 898aef5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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"},