Skip to content

Commit 5f2384f

Browse files
authored
Patch protobuf for PT sagemaker training (#4967)
* upgrade protobuf * upgrade sagemaker-trainin * pin protobuf6 * revert toml
1 parent 21e2b8b commit 5f2384f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

miscellaneous_scripts/dockerfile_patch_script.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,9 @@ fi
4343
chmod +x $PATCHING_INFO_PATH/patch-details/install_script_language.sh && \
4444
$PATCHING_INFO_PATH/patch-details/install_script_language.sh
4545

46-
# Upgrade sagemaker-training
46+
# Upgrade sagemaker-training
4747
if [[ $LATEST_RELEASED_IMAGE_URI =~ ^763104351884\.dkr\.ecr\.us-west-2\.amazonaws\.com/pytorch-training:2\.[4-6](.+)sagemaker ]]; then
48-
pip install "sagemaker-training>4.7.4,<5" --upgrade
48+
pip install -U "sagemaker-training>4.7.4" "protobuf>=4.25.8,<6"
4949
fi
5050

5151
# For PT inference sagemaker images, replace torchserve-entrypoint.py with the latest one

0 commit comments

Comments
 (0)