diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py index 2d22a45359..1f06036262 100644 --- a/src/sagemaker/model.py +++ b/src/sagemaker/model.py @@ -574,6 +574,7 @@ def _upload_code(self, key_prefix: str, repack: bool = False) -> None: script=self.entry_point, directory=self.source_dir, dependencies=self.dependencies, + kms_key=self.model_kms_key, settings=self.sagemaker_session.settings, )