diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py index 533a427747..2bdea34d6f 100644 --- a/src/sagemaker/huggingface/model.py +++ b/src/sagemaker/huggingface/model.py @@ -533,6 +533,7 @@ def prepare_container_def( deploy_image, self.repacked_model_data or self.model_data, deploy_env, + image_config=self.image_config, accept_eula=accept_eula, model_reference_arn=model_reference_arn, )