We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 21e2b8b commit 5f2384fCopy full SHA for 5f2384f
miscellaneous_scripts/dockerfile_patch_script.sh
@@ -43,9 +43,9 @@ fi
43
chmod +x $PATCHING_INFO_PATH/patch-details/install_script_language.sh && \
44
$PATCHING_INFO_PATH/patch-details/install_script_language.sh
45
46
-# Upgrade sagemaker-training
+# Upgrade sagemaker-training
47
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
+ pip install -U "sagemaker-training>4.7.4" "protobuf>=4.25.8,<6"
49
fi
50
51
# For PT inference sagemaker images, replace torchserve-entrypoint.py with the latest one
0 commit comments