From fec3b8800e09bafd553ff0c2b0ead63dbd56424a Mon Sep 17 00:00:00 2001
From: Nate Brake <33383515+njbrake@users.noreply.github.com>
Date: Wed, 22 May 2024 15:02:32 -0400
Subject: [PATCH] Update model.py

Allow for the container to be passed the image_config
---
 src/sagemaker/huggingface/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py
index f71dca0ac8..66881b1381 100644
--- a/src/sagemaker/huggingface/model.py
+++ b/src/sagemaker/huggingface/model.py
@@ -523,6 +523,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,
         )