Add torchrec to promote s3 script (#1680)

* Add torchrec to promote s3 script * Add torchrec version to release_version.sh
pytorch · Jan 23, 2024 · 122ff0d · 122ff0d
1 parent 896b6df
commit 122ff0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/release/promote.sh b/release/promote.sh
@@ -11,6 +11,7 @@ TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.1}
 TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1}
 TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.1}
 TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1}
+TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0}
 
 DRY_RUN=${DRY_RUN:-enabled}
 
@@ -102,6 +103,7 @@ promote_pypi() {
 # promote_s3 torchaudio whl "${TORCHAUDIO_VERSION}"
 # promote_s3 torchtext whl "${TORCHTEXT_VERSION}"
 # promote_s3 torchdata whl "${TORCHDATA_VERSION}"
+# promote_s3 torchrec whl "${TORCHREC_VERSION}"
 # promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}"
 
 # promote_conda torchtriton conda "2.1.0"

diff --git a/release/release_versions.sh b/release/release_versions.sh
@@ -6,3 +6,4 @@ TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.1}
 TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1}
 TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.1}
 TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1}
+TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0}