Update PT pin to 2.6 for modules/_export (#2336)

pytorch · Feb 3, 2025 · 059cad9 · 059cad9
1 parent 6487029
commit 059cad9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/torchtune/modules/_export/install_requirements.sh b/torchtune/modules/_export/install_requirements.sh
@@ -5,7 +5,5 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-NIGHTLY_VERSION="dev20241121"
-
-# Install pytorch nightly for export-friendly modules to run.
-pip install torch==2.6.0.${NIGHTLY_VERSION} torchvision==0.20.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu
+# Install pytorch
+pip install torch==2.6.0 torchvision==0.21.0 --extra-index-url https://download.pytorch.org/whl/nightly/cpu